龙空技术网

Python中DataFrame判断两列数据是否相等的方法

Deities冥 207

前言:

如今姐妹们对“python比较两列数据”可能比较关注,你们都想要学习一些“python比较两列数据”的相关文章。那么小编同时在网摘上收集了一些关于“python比较两列数据””的相关知识,希望朋友们能喜欢,朋友们一起来了解一下吧!

数据准备import numpy as npimport pandas as pdimport jsonimport psycopg2

data = {'A':[1,2,3,4,'hello','world'],'B':[1,2,3,7,'hello','word']}df_data = pd.DataFrame(data = data)df_data

A

B

0

1

1

1

2

2

2

3

3

3

4

7

4

hello

hello

5

world

word

方法一:写函数判断# 方法一:写函数判断# 判断是否相等def is_equal_or_not(a,b):if a == b:return 1else:return 0# 数据处理df_data['AB列数据是否相等'] = df_data.apply(lambda x : is_equal_or_not(x['A'],x['B']),axis = 1)df_data

A

B

AB列数据是否相等

0

1

1

1

1

2

2

1

2

3

3

1

3

4

7

0

4

hello

hello

1

5

world

word

0

方法二:直接判断# 方法二:直接判断df_data['AB列数据是否相等2'] = (df_data['A'] == df_data['B'])*1df_data

A

B

AB列数据是否相等

AB列数据是否相等2

0

1

1

1

1

1

2

2

1

1

2

3

3

1

1

3

4

7

0

0

4

hello

hello

1

1

5

world

word

0

0

标签: #python比较两列数据