python - Pandas Dataframe 查找所有列相等的行

Question

我有一个包含字符的数据框 - 我想要一个逐行的 bool 结果，告诉我该行的所有列是否具有相同的值。

例如，我有

df = [  a   b   c   d

0  'C'   'C'   'C'   'C' 

1  'C'   'C'   'A'   'A'

2  'A'   'A'   'A'   'A' ]

我希望结果是

0  True

1  False

2  True

我试过 .all 但似乎我只能检查 all 是否等于一个字母。我能想到的唯一另一种方法是在每一行上做一个唯一的，看看它是否等于 1？提前致谢。

Answer 1

我认为最简洁的方法是使用 eq: 对照第一列检查所有列:

In [11]: df
Out[11]: 
   a  b  c  d
0  C  C  C  C
1  C  C  A  A
2  A  A  A  A

In [12]: df.iloc[:, 0]
Out[12]: 
0    C
1    C
2    A
Name: a, dtype: object

In [13]: df.eq(df.iloc[:, 0], axis=0)
Out[13]: 
      a     b      c      d
0  True  True   True   True
1  True  True  False  False
2  True  True   True   True

现在你可以使用 all(如果它们都等于第一项，则它们都相等):

In [14]: df.eq(df.iloc[:, 0], axis=0).all(1)
Out[14]: 
0     True
1    False
2     True
dtype: bool