您的位置:首页 > 技术中心 > 其他 >

Python中DataFrame怎么判断两列数据是否相等

时间:2023-05-19 18:50

数据准备

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

AB
011
122
233
347
4hellohello
5worldword

方法一:写函数判断

# 方法一:写函数判断# 判断是否相等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

ABAB列数据是否相等
0111
1221
2331
3470
4hellohello1
5worldword0

方法二:直接判断

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

ABAB列数据是否相等AB列数据是否相等2
01111
12211
23311
34700
4hellohello11
5worldword00

以上就是Python中DataFrame怎么判断两列数据是否相等的详细内容,更多请关注Gxl网其它相关文章!

本类排行

今日推荐

热门手游