Pandas 中的部分查询

Question

在 Pandas 中，可以执行以下操作：

df.query("some_column in ('foo', 'bar')")

它检索df 的子集，其中some_column 列的值为'foo' 或'bar'。

但是如果我想反其道而行之呢？例如，查找df 的子集，其中some_column 列中字符串的任何子字符串 匹配“foo”或“bar”。

如何在 Pandas 中做到这一点？

Answer 1

使用string methods。

df['some column'].str.contains(r'foo|bar')

例如：

In [8]: Series(['foo', 'bar', 'fo', 'foo2']).str.contains(r'foo|bar')
Out[8]: 
0     True
1     True
2    False
3     True
dtype: bool