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

寻技术 Python编程 2023年07月11日 186

本文小编为大家详细介绍“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

读到这里,这篇“Python中DataFrame怎么判断两列数据是否相等”文章已经介绍完毕,想要掌握这篇文章的知识点还需要大家自己动手实践使用过才能领会,如果想了解更多相关内容的文章,欢迎关注***行业资讯频道。

关闭

用微信“扫一扫”