按子字符串列表过滤数据帧 [英] Filtering DataFrame by list of substrings

问题描述

基于此答案，有没有办法通过子字符串列表过滤 Pandas 数据框?

Building off this answer, is there a way to filter a Pandas dataframe by a list of substrings?

假设我想查找 df['menu_item'] 包含 fresh 或 spaghetti

Say I want to find all rows where df['menu_item'] contains fresh or spaghetti

没有这样的东西:

df[df['menu_item'].str.contains('fresh') |(df['menu_item'].str.contains('spaghetti')]

推荐答案

str.contains 您使用的方法接受正则表达式，因此使用正则表达式 | 作为或:

df[df['menu_item'].str.contains('fresh|spaghetti')]

示例输入:

          menu_item
0        fresh fish
1      fresher fish
2           lasagna
3     spaghetti o's
4  something edible

示例输出:

       menu_item
0     fresh fish
1   fresher fish
3  spaghetti o's

这篇关于按子字符串列表过滤数据帧的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！