Maison  >  Article  >  développement back-end  >  Comment déterminer si deux colonnes de données sont égales dans DataFrame en Python

Comment déterminer si deux colonnes de données sont égales dans DataFrame en Python

王林
王林avant
2023-05-19 15:49:062448parcourir

Préparation des données

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 bonjour bonjour
5 monde mot

Méthode 1 : Écrire le jugement de fonction

# 方法一:写函数判断
# 判断是否相等
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
Méthode 2 : Déterminer directement si les données de la colonne AB sont égales 1 1 112211

A B Les données de la colonne AB sont-elles égales
2 3 3 1
3 4 7 0
4 bonjour bonjour 1
5 monde mot 0

234bonjourmonde

3
1 1 3
7 0 0 4
bonjour 1 1 5
mot 0 0

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!

Déclaration:
Cet article est reproduit dans:. en cas de violation, veuillez contacter admin@php.cn Supprimer