Python中如何通过字符串参数动态访问DataFrame列并绘图？

当你需要用字符串类型的列名调用DataFrame的plot方法时，直接用方括号索引替代属性访问即可，也就是df[target].plot()——这是Pandas中通过字符串列名访问列的标准方式，完美适配你传入字符串参数的场景。

完整函数示例 #

import pandas as pd

def plot_target(df, target):
    # 可选：先校验列是否存在，避免报错
    if target not in df.columns:
        raise ValueError(f"列 '{target}' 不存在于当前DataFrame中")
    # 核心：用方括号索引字符串列名，再调用plot
    ax = df[target].plot()
    # 可选：给图表加标题，提升可读性
    ax.set_title(f"{target} 数据趋势图")
    return ax

关键说明 #

• df.target1属于属性访问，它要求列名是合法的Python标识符（不能有空格、特殊字符，也不能和Pandas内置方法重名），无法直接用字符串变量替代。
• df[target]是索引访问，支持任意合法的字符串列名，完全满足你通过参数传递列名的需求。

使用示例 #

假设你的DataFrame结构如下：

data = {
    "var1": [1, 3, 5, 7],
    "var2": [2, 4, 6, 8],
    "target1": [10, 20, 30, 40],
    "target2": [15, 25, 35, 45]
}
df = pd.DataFrame(data)

# 调用函数绘制target2的图表
plot_target(df, "target2")

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