本文小编为大家详细介绍“Python中DataFrame怎么判断两列数据是否相等”，内容详细，步骤清晰，细节处理妥当，希望这篇“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 |