Pandas 判断两列内容是否相同并将结果新建一列

1、首先写一个函数,实现判断是否相等的功能,如下:

def function(a, b):

if a == b:

return 1

else:

return 0

2、用 pandas 里面强大的 lambda 处理文件(如CSV文件):

import pandas as pd;

data = pd.read_csv('test.csv')

df = pd.DataFrame(data)

df['bool'] = df.apply(lambda x : function(x['A'],x['B']),axis = 1)

其中 A, B是需要比较是否相等的两列的列名。

以上,新生成的 bool 列内容即是 0 和 1 用来表示 A 列和 B列是否相等。

————————————————

版权声明:本文为CSDN博主「程序媛三妹」的原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接及本声明。

原文链接:https://blog.csdn.net/u011675334/article/details/100782022

推荐阅读 更多精彩内容