Pandas读取CSV创建DataFrame并筛选指定列值为yes的行的方法

前置说明 #

假设你需要筛选的列名为target_col，实际使用时替换为你自己的列名，同时将文件路径替换为你本地CSV的真实路径。

完整实现步骤 #

• 导入Pandas库

import pandas as pd

• 读取CSV生成原始DataFrame

raw_df = pd.read_csv("your_file_path.csv")

• 筛选目标列取值为yes的行，生成结果DataFrame

写法1：布尔索引（最常用） #

target_df = raw_df[raw_df["target_col"] == "yes"]

写法2：query方法（语法更简洁） #

target_df = raw_df.query("target_col == 'yes'")

异常场景处理 #

如果CSV中目标列的取值可能存在首尾空格（比如 yes、yes 这类异常值），可以先对列值做清洗再筛选：

target_df = raw_df[raw_df["target_col"].str.strip() == "yes"]

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