Python中DataFrame怎么判断两列数据是否相等
时间:2023-05-19 18:50
以上就是Python中DataFrame怎么判断两列数据是否相等的详细内容,更多请关注Gxl网其它相关文章!数据准备
import numpy as npimport pandas as pdimport jsonimport psycopg2data = { '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'])*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