按关键字列表和字符串列过滤数据框

时间:2018-02-02 14:01:38

标签: python pandas dataframe

我有一个这样的数据框:

Name     Street
Tom      125, Fullton Street
Bill     133, Brighton Street
Simon    152, Lampton Street
Angie    89,  Langdon Street

现在我想通过使用街道名称列表来保留那些属于某个社区的行:

neighbours = [Fullton, Brighton] 

我使用了这段代码,但它只保留了Fullton street的行:

df_neighbours = df[df['Street'].str.contains(neighbours)]

有什么想法吗? 谢谢!

1 个答案:

答案 0 :(得分:3)

您非常接近 - 需要加入|加入正则表达式or

df_neighbours = df[df['Street'].str.contains('|'.join(neighbours))]