Сравните два кадра данных по каждому столбцу в python? - PullRequest
0 голосов
/ 03 мая 2020

Df1:

Roll_No Sub1 Sub2 Sub3
1001      40  50   60
1002      65  87   30
1003      36  49   40
1004      29  83   50

Я хочу сравнить Roll_No 1001 с каждым Roll_No на подуровне. каким-то образом я создал df2, который выглядит следующим образом: df2

Roll_No Sub1 Sub2 Sub3
1001      40  50   60
1001      40  50   60
1001      40  50   60
1001      40  50   60

Теперь я хочу сравнить каждый столбец с df1 & df2 в нескольких условиях.

Cond1 - df1.Sub1 -df2.Sub1 > 5
then 1 else 0
Cond2 - df1.Sub2 -df2.Sub2 > 10 
then 2 else 0
Cond3 - df1.Sub3 -df2.Sub3 > 7 
then 3 else 0

Вывод:

Roll_No Sub1 Sub2 Sub3
1001      0    0   0
1002      0    2   3
1003      0    0   3
1004      1    0   3

1 Ответ

2 голосов
/ 03 мая 2020
import pandas as pd
import numpy as np

Output = df1.copy()

Output['Sub1'] = np.where(df1['Sub1'] - df2['Sub1'] > 5, 1, 0)
Output['Sub2'] = np.where(df1['Sub2'] - df2['Sub2'] > 10, 2, 0)
Output['Sub3'] = np.where(df1['Sub3'] - df2['Sub3'] > 7, 3, 0)

Источник https://datatofish.com/compare-values-dataframes/

...