python - 在选择列 Pandas 中选择值相同的行？

Question

我正在尝试获取我的数据框的子集，该子集对任意数量的列具有相同的值。

有点像

df[df['col1'] == df['col2'] ==....] 

但对于任意 数量的列名。

Answer 1

我觉得应该可以

columns_to_compare = ["foo", "bar"]

def check_all_equal(iterator):
   return len(set(iterator)) <= 1

df[df[columns_to_compare].apply(lambda row: check_all_equal(row), axis=1)]