Bagaimana untuk menentukan sama ada dua lajur data adalah sama dalam DataFrame dalam Python

Penyediaan data

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

Kaedah satu: tulis penghakiman fungsi

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

Kaedah kedua: hakim terus

# 方法二：直接判断
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

Atas ialah kandungan terperinci Bagaimana untuk menentukan sama ada dua lajur data adalah sama dalam DataFrame dalam Python. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!