网络编程
位置:首页>> 网络编程>> Python编程>> Python中DataFrame判断两列数据是否相等的方法

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

作者:大地之灯  发布时间:2023-12-09 10:17:37 

标签: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

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'])*1
df_data

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

来源:https://blog.csdn.net/qq_33489955/article/details/129237660

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com