如何基于列名含多个指定字符串的条件筛选Pandas DataFrame的列
filter的like参数仅支持传入单个字符串做模糊匹配,无法直接传入多个关键词实现多条件匹配,你可以用以下两种方法实现需求:
方法1:使用filter的regex参数(最简便)
利用正则的或逻辑|拼接多个匹配关键词,代码如下:
import pandas as pd data = {'Name': ['Jai', 'Princi', 'Gaurav', 'Anuj'], 'Height of Person': [5.1, 6.2, 5.1, 5.2], 'Qualification': ['Msc', 'MA', 'Msc', 'Msc'], 'Country is': ['US', 'UK', 'GE', 'ET'] } df = pd.DataFrame(data) # 匹配包含Name或Country的列 df2 = df.filter(regex='Name|Country') display(df2)
说明:regex参数传入的正则表达式中,|代表或逻辑,只要列名包含任意一个关键词就会被保留。如果需要不区分大小写匹配,可以修改正则为regex='(?i)name|country'。
方法2:使用列表推导式实现(可读性更高,适合多关键词场景)
如果需要匹配的关键词较多,不想拼接正则,可以先定义关键词列表,再遍历列名筛选:
# 定义需要匹配的关键词列表 match_keys = ['Name', 'Country'] # 筛选列名中包含任意一个关键词的列 df2 = df[[col for col in df.columns if any(key in col for key in match_keys)]] display(df2)
两种方法运行后都可得到预期的输出结果。
内容的提问来源于stack exchange,提问作者xxgaryxx
相关产品推荐
相关产品推荐

