如何比较pandas、python中的列行

Question

我得到了熊猫数据框，看起来是这样的：

    HomeTeam    AwayTeam    HTR FTR

39  Arsenal     Tottenham   2   12
136 Norwich     Arsenal     1   2
101 Arsenal     Wolves      1   1

我想在那里添加一列，以比较每一行中的HTR和FTR。有3个可能的值：1、2或12

如果HTR=FTR，我希望在新列中得到“1”， 如果高温气冷堆≠ FTR我想得到“0”

Answer 1

让我们生成一些人工数据，如您所述：

df = pd.DataFrame({'HTR': np.random.choice([1,2,12],100),
                   'FTR':np.random.choice([1,2,12],100)})

Dataframe现在看起来像这样：

    HTR FTR

39   2  12
136  1  2
101  1  1

现在，让我们来解决您的问题：

df["are_same"] = (df["HTR"]==df["FTR"]).astype(int)

添加新列后，Dataframe如下所示：

    HTR FTR are_same

39   2  12  0
136  1  2   0
101  1  1   1