Rumah  >  Artikel  >  pembangunan bahagian belakang  >  Bagaimana untuk menentukan sama ada dua lajur data adalah sama dalam DataFrame dalam Python

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

王林
王林ke hadapan
2023-05-19 15:49:062451semak imbas

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!

Kenyataan:
Artikel ini dikembalikan pada:yisu.com. Jika ada pelanggaran, sila hubungi admin@php.cn Padam