使用pd.read_csv动态指定列名读取数据失败，硬编码正常求排查

我之前也碰到过一模一样的情况——手动输入列名就读取不到，硬编码就正常，多半是这几个坑在搞鬼，咱们一步步排查：

1. 最常见的原因：输入列名和CSV实际列名不匹配 #

手动输入很容易出现拼写错误、大小写不一致或者多余空格的问题，比如CSV里的列名是User_Age，你输入成user_age或者User Age（多了空格），pandas就认不出来了。

先确认CSV里的真实列名，用这段代码打印出来：

# 仅读取跳过前两行后的表头，不加载数据
df_header = pd.read_csv(file, skiprows=[0,1], nrows=0)
print("Available columns:", df_header.columns.tolist())

把打印出来的列名和你输入的对比，就能立刻发现问题。

另外，给输入的列名做个清洗，去掉前后可能不小心输入的空格：

col_name = raw_input("Enter column name: ").strip()

2. 容易忽略的坑：names参数的误用 #

看你的代码里加了name=cols_used（应该是names，你可能打错了），这个参数会覆盖CSV里的原有表头，用你指定的列表作为新列名。

• 如果跳过前两行后，第三行是CSV的真实表头：那你根本不需要names参数！这时候用usecols会根据CSV的表头去匹配列，而names会把第三行当成数据行，导致列名完全错位。
• 如果确实需要自定义列名：那usecols应该用列的索引位置，而不是列名，比如你要读第0列和第3列：

  read_cols = pd.read_csv(file, usecols=[0, 3], skiprows=[0,1], names=cols_used)
  

修正后的基础代码（假设第三行是表头）：

file = some_file
col_name = raw_input("Enter column name: ").strip()
cols_used = ["X", col_name]
# 去掉names参数，让pandas用CSV自带的表头匹配列
read_cols = pd.read_csv(file, usecols=cols_used, skiprows=[0,1])
test = pd.unique(read_cols["X"])

3. 进阶优化：给用户输入做校验 #

可以在读取前先检查输入的列名是否存在，避免无效输入：

file = some_file
# 先获取所有可用列名
df_header = pd.read_csv(file, skiprows=[0,1], nrows=0)
all_cols = df_header.columns.tolist()

col_name = raw_input("Enter column name: ").strip()
# 循环直到用户输入正确的列名
while col_name not in all_cols:
    print(f"Oops, column '{col_name}' doesn't exist! Available columns are: {all_cols}")
    col_name = raw_input("Please enter a valid column name: ").strip()

cols_used = ["X", col_name]
read_cols = pd.read_csv(file, usecols=cols_used, skiprows=[0,1])
test = pd.unique(read_cols["X"])

按照上面的步骤排查，应该就能解决问题啦！

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