python - 在 Pandas Dataframe 的多列(不是所有列)中查找具有相同值的行

Question

下面是我的数据框:

X1  X2  X3  X4  X5
A   B   C   10  BAM
A   A   A   12  BAM
B   B   B   10  BAM
A   B   B   60  BAM

我希望那些行在列(X1, X2,X3) 中具有相同的值。在这里我们可以看到 2nd 和 3rd 行的上述 3 列具有相同的值。 我想要的输出是:

 X1 X2  X3  X4  X5
A   A   A   12  BAM
B   B   B   10  BAM

我试过如下:

yourdf1=df[df.nunique(0)==0]
print(yourdf1)

但是这里我遇到了一个错误。谁能帮帮我。

Answer 1

通过 axis=1 在 DataFrame.nunique 中选择列表中的列以测试每行的唯一值数量并测试 1 以过滤 boolean indexing :

yourdf1 = df[df[['X1','X2','X3']].nunique(axis=1) == 1]
print(yourdf1)
  X1 X2 X3  X4   X5
1  A  A  A  12  BAM
2  B  B  B  10  BAM

另一个解决方案是使用 DataFrame.eq使用过滤后的 DataFrame，按第一列进行比较并按 DataFrame.all 获取每行的所有真值:

df1 = df[['X1','X2','X3']]
yourdf1 = df[df1.eq(df1.iloc[:, 0], axis=0).all(axis=1)]
print(yourdf1)

  X1 X2 X3  X4   X5
1  A  A  A  12  BAM
2  B  B  B  10  BAM