python - Pandas 将聚合与原始列进行比较

Question

我很难将 pandas 数据框与从该数据框生成的聚合列进行比较。具体来说，如果我有

df = Col1 Col2
     a    p
     a    q
     a    r
     b    r
     c    s

我想要一个 bool 数据框，它的每个条目都等于该列的模态值。我试过了

df == df.mode()

但这给了我只能比较标记相同的DataFrame对象。我想要一些可以在第一个数据帧的行上广播第二个数据帧来计算结果的东西。需要明确的是，在上述情况下我有

modal_df = df.mode()
modal_df = Col1 Col2
           a    r

desired_result = Col1  Col2
                 True  False
                 True  False
                 True  True
                 False True
                 False False

Answer 1

您可以将df与values创建的numpy array进行比较:

print (df.mode().values)
[['a' 'r']]

print (df == df.mode().values)
    Col1   Col2
0   True  False
1   True  False
2   True   True
3  False   True
4  False  False