数据准备
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 |
|---|
到此这篇关于DataFrame判断两列数据是否相等的方法的文章就介绍到这了,更多相关DataFrame判断数据相等内容请搜索编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程网！
免责声明：
① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的，并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据，供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。
② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341
