Pandas中的部分查询

Question

在熊猫中，人们可以做到以下几点：

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

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

但如果我想做相反的事情呢？例如，找到df的子集，some_column列中字符串的任何子字符串匹配＆＃39; foo＆＃39;或者＆＃39; bar`。

我怎么能在熊猫中做到这一点？

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