pandas中如何通过变量值动态引用对应名称的DataFrame列

核心原则：不要尝试动态拼接变量名做引用，用字典映射是最稳妥、可维护性最高的方案

• 不推荐做法说明
  不要用globals()[f"{i}_df"]这类直接读取全局/局部变量表的写法，这种方式没有任何校验，一旦输入值超出预期很容易调用到无关变量，出问题极难排查，也无法通过代码静态检查。

• 标准实现步骤

  1. 定义一个字典作为DataFrame的存储容器，把你原来分支里构造的DataFrame直接存入字典，key就用i对应的可选值a/b

  import pandas as pd
  
  df_map = {}
  # 保留你原有的if分支逻辑，只是把赋值目标改成字典对应key
  if 条件A:
      # 原来写a_df = xxx的逻辑，直接赋值给df_map['a']
      df_map['a'] = pd.DataFrame(...)
  if 条件B:
      # 原来写b_df = xxx的逻辑，直接赋值给df_map['b']
      df_map['b'] = pd.DataFrame(...)
  

  如果你不想修改原有分支里a_df/b_df的定义代码，也可以在两个分支执行完之后再组装映射字典：

  # 原有分支逻辑完全不动
  if 条件A:
      a_df = pd.DataFrame(...)
  if 条件B:
      b_df = pd.DataFrame(...)
  
  # 分支结束后统一组装映射
  df_map = {
      "a": a_df,
      "b": b_df
  }
  

  2. 拿到用户输入的i之后，直接通过字典key取对应DataFrame，再访问col列即可，比你预期的{i}_df['col']写法更简洁：

  i = input("请输入a/b选择对应数据集：").strip()
  # 可选：加个输入校验，避免非法输入报错
  if i not in df_map:
      raise ValueError("输入值仅支持a或b")
  
  # 直接取目标列
  target_col_data = df_map[i]['col']
  

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