熊猫选择并写入包含某些文本的行

Question

我只想在包含特定列"col"文本的数据框中保留行。 在此示例中， "WORD1"或"WORD2" 。

df = df["col"].str.contains("WORD1|WORD2")
df.to_csv("write.csv")

这将返回True或False 。 但是，如何使它写入与这些critera匹配的整个行，而不仅仅是呈现布尔值？

Answer 1

返回的是一个布尔系列，您可以使用它来过滤df：

df = df[df["col"].str.contains("WORD1|WORD2")]

然后，您可以照常将其写出：

df.to_csv("write.csv")

例：

In [14]:    
df = pd.DataFrame({'col':['word', 'WORD1', 'WORD2', 'WORD3']})
df

Out[14]:
     col
0   word
1  WORD1
2  WORD2
3  WORD3

In [15]:    
df[df['col'].str.contains('WORD1|WORD2')]

Out[15]:
     col
1  WORD1
2  WORD2