Pandas如何多条件筛选DataFrame后调用tolist()将列转为列表

最简实现（适配当前需求） #

筛选某一列匹配多个目标值的场景，直接使用isin()方法传入待匹配的值列表做布尔索引，无需写冗余的或逻辑，也不需要循环，代码如下：

import pandas as pd

# 构造测试数据
data = {'file_type':['excel', 'csv', 'xls', 'csv'],
        'file_name':['my_files.xls', 'txt_file.csv', 'test_file.xls', 'new_file.csv'],
        'oter_col':['other_value1', 'other_value2', 'other_value3', 'other_value4']}
df = pd.DataFrame(data)

# 筛选file_type为excel、xls的行，提取file_name列转为列表
result = df[df['file_type'].isin(['excel', 'xls'])]['file_name'].tolist()

运行后result的输出为预期结果：['my_files.xls', 'test_file.xls']

多条件筛选避坑说明 #

之前用or/and写条件运行失败，是因为Pandas对DataFrame/Series的布尔逻辑运算有专属规则：

• 同列匹配多个固定值优先用isin()，写法最简洁，执行效率更高
• 涉及不同列的组合逻辑判断时，不能直接用Python原生的and/or关键字，要使用按位运算符：
  • 逻辑与（同时满足）用&
  • 逻辑或（满足任意一个）用|
  • 每个独立的判断条件必须用圆括号包裹，否则会因为运算符优先级报错
    示例（筛选file_type为csv且文件名包含new的记录）：

  demo_result = df[(df['file_type'] == 'csv') & (df['file_name'].str.contains('new'))]['file_name'].tolist()
  # 输出为 ['new_file.csv']
  

避坑提示：直接写df['file_type']=='excel' or df['file_type']=='xls'会触发报错，因为Pandas不支持用Python原生and/or处理序列级布尔运算，必须用&/|运算符，且每个子条件必须单独用圆括号包裹，避免运算符优先级导致的逻辑错误。

内容的提问来源于stack exchange，提问作者Ruban Bob