Python Pandas使用变量获取DataFrame列时触发KeyError问题求助

提取月份与读取文件的代码 #

# 提取月份值
month = str(DF['COLUMN'].max())[6:7]

# 读取目标Excel文件
file = pandas.read_excel('myfile.xlsx', sheet_name='sheet1',engine='openpyxl')
mycolumns= file.columns.values.tolist()

循环匹配月份变量的代码 #

for i in mycolumns:
    if i == month :
        this_month = i

触发错误的操作与报错信息 #

执行以下代码尝试获取对应列的所有值时：

file.loc[:,f'{this_month}']

触发KeyError：

KeyError:'VariableName'

已尝试的无效操作 #

• 将列名转为列表、数组
• 去除列名空格
• 转换month变量的类型（str与int互转）

示例数据格式 #

与实际数据格式一致的示例DataFrame：

import pandas as pd
data = {'1' : ['50000','70000','90000','872'],
    '2' : ['8394' , '83704','94809','02984'],
    '3' : ['3664' , '74637','39847','38984']
    }
df   = pd.DataFrame(data) 

• 先确认变量与列名的实际内容
  直接打印month和mycolumns的原始表示（用repr(month)和[repr(col) for col in mycolumns]），排查是否存在隐形字符（如换行符、制表符）、类型不匹配（比如列名是int、month是str）或字符差异。

• 简化匹配逻辑，避免循环遗漏
  循环如果没找到匹配项，this_month会处于未定义或保留旧值的状态，改用存在性判断更可靠：

  if month in mycolumns:
      this_month = month
  else:
      print(f"月份值 {month} 不在列名列表中")
      print("当前列名：", mycolumns)
  

• 直接用month索引，跳过中间变量
  确认month有效后，无需额外赋值this_month，直接索引：

  if month in file.columns:
      target_data = file[month]
  else:
      print(f"列名 {month} 不存在于Excel文件中")
  

• 统一列名与变量的类型
  Excel读取的列名可能是int类型，而提取的month是str，此时==判断不成立，可统一转换类型：

  # 把列名全部转为字符串
  mycolumns = [str(col) for col in file.columns]
  # 或者把month转为整数（如果月份是纯数字）
  month = int(str(DF['COLUMN'].max())[6:7])
  

• 清理隐形字符
  如果列名或month存在空格、不可见字符，用strip()处理：

  # 处理列名的隐形字符
  mycolumns = [str(col).strip() for col in file.columns]
  # 同时处理month变量
  month = str(DF['COLUMN'].max())[6:7].strip()
  

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