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

目录

• 数据准备
• 方法一：写函数判断
• 方法二：直接判断

数据准备

import numpy as np
import pandas as pd
import json
import 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 1
    else:
        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'])*1
df_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

 到此这篇关于DataFrame判断两列数据是否相等的方法的文章就介绍到这了,更多相关DataFrame判断数据相等内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！