关于Pandas DataFrame.assign()方法列名传参的技术咨询

为什么assign()需要传入不带引号的列名？ #

这是因为assign()方法是通过Python关键字参数来识别列名的。当你写df.assign(Grade='Total')时，Grade在这里是一个关键字参数，Python会自动将它转换为字符串键'Grade'传递给方法内部的参数集合。这种设计是为了让代码更简洁直观，符合Python的语法习惯——关键字参数本身就要求是合法的标识符（不能有空格、特殊字符等，也不能直接用字符串）。

怎么规避“不能用变量指定列名”的限制？ #

当然有办法！如果你想用变量动态指定列名，可以借助字典解包的方式来实现。具体来说：

• 先创建一个字典，把你的列名字变量作为键，对应的值作为列的新内容
• 然后用**操作符把这个字典解包成assign()的关键字参数

修正你的示例代码 #

看你提供的dummy函数，问题出在直接把变量_g当作了关键字参数名，而不是用它的值作为列名。修改后的代码如下：

import pandas as pd

df = pd.DataFrame(columns=['Grade'])
df['Grade'] = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]
df_temp = df.assign(Grade='Total')

def dummy(df, _g):
    # 使用字典解包，将_g变量的值作为列名
    return df.assign(**{_g: 'Total'})  

df_temp = dummy(df, 'Grade')
print(df_temp)

运行这段代码后，Grade列会被正确修改为'Total'，而不会创建新的_g列。

补充说明 #

除了字典解包，你也可以考虑直接使用传统的列赋值方式（比如df[_g] = 'Total'），但如果想保持链式调用的风格（这也是assign()的常用场景），字典解包是最适合的方案。

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