在给定字符串条件的情况下删除熊猫中的数据

Question

鉴于以下情况，我无法理解此处的机制。

我有一个 dataframe 从 .csv 读取：

  a1 b1 c1
1 aa bb cc
2 ab ba ca 

df.drop(df['a1'].str.contains('aa',case = False))

我想删除列 a1 中包含 'aa' 的所有行

我相信已经尝试了这里的一切，但仍然得到：

ValueError: 标签 [False False False ... False False False] 不包含在轴中

是的，我也试过了

skipinitialspace=True
axis=1

任何帮助将不胜感激，谢谢。

Answer 1

str.contains 返回一个掩码：

df['a1'].str.contains('aa',case = False)

1     True
2    False
Name: a1, dtype: bool

但是，drop 接受索引标签，而不是布尔掩码。如果您打开drop 上的帮助，您可能会亲眼目睹这一点：

?df.drop

Signature: df.drop(labels=None, axis=0, index=None, columns=None, level=None, inplace=False, errors='raise')
Docstring:
Return new object with labels in requested axis removed.

Parameters
----------
labels : single label or list-like
    Index or column labels to drop.

您可以从掩码中找出索引标签并将 那些 传递给 drop

idx = df.index[df['a1'].str.contains('aa')]
df.drop(idx)

   a1  b1  c1
2  ab  ba  ca

但是，这风太大了，所以我建议只坚持基于条件删除行的 pandaic 方法，布尔索引：

df[~df['a1'].str.contains('aa')]

   a1  b1  c1
2  ab  ba  ca

如果有人有兴趣删除列表中包含字符串的行

df = df[~df['a1'].str.contains('|'.join(my_list))]

确保去掉空格。感谢https://stackoverflow.com/a/45681254/9500464