删除给定字符串条件的熊猫中的数据

Question

考虑到以下情况，我很难理解这里的机制。

我有一个数据帧从读取。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]轴中不包含

是的，我也试过

skipinitialspace=True
axis=1

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

Answer 1

str.contains返回掩码：

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

1     True
2    False
Name: a1, dtype: bool

但是，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.

您可以从掩码中找出索引标签，并将其传递给删除

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