python - 如果行在 Pandas 中包含特定值，则删除列

Question

我开始学习 Pandas。我在 SO 中看到了很多问题，人们询问如果列与特定值匹配如何删除行。

我的情况恰恰相反。想象一下有这个数据框:

您想知道的是，如果任何列在其任何行中具有值 salty，则应删除该列，结果是:

我尝试过与此有几个相似之处:

if df.loc[df['A'] == 'salty']:
   df.drop(df.columns[0], axis=1, inplace=True)

但是我完全找不到关于如何根据该列的行值删除列的文档。该代码是查找特定列和始终删除第一列的混合体(因为我的想法是在 for 循环中的所有列中搜索该列中一行的值。

Answer 1

对您的值进行比较，然后使用 DataFrame.any 获取要索引的掩码:

df.loc[:, ~(df == 'Salty').any()]

如果你坚持使用drop，这就是你需要做的。传递索引列表:

df.drop(columns=df.columns[(df == 'Salty').any()])

---

df = pd.DataFrame({
    'A': ['Mountain', 'Salty'], 'B': ['Lake', 'Hotty'], 'C': ['River', 'Coldy']})
df
          A      B      C
0  Mountain   Lake  River
1     Salty  Hotty  Coldy

(df == 'Salty').any()
A     True
B    False
C    False
dtype: bool

df.loc[:, ~(df == 'Salty').any()]
       B      C
0   Lake  River
1  Hotty  Coldy

df.columns[(df == 'Salty').any()]
# Index(['A'], dtype='object')

df.drop(columns=df.columns[(df == 'Salty').any()])
       B      C
0   Lake  River
1  Hotty  Coldy