Python函数传入变量，用作函数内的名称

Question

我想更改数据框的列名：

def test(var):
    dic={'a': [5, 9],
     'b': [4, 1],
     'c': [2, 9]}

    df=pd.DataFrame(dic).transpose()
    df.columns=['aad',var_'k']
    return df
test('lalala')   

我希望第二列名称为var_'k';也就是说，如果我传入lalala，则列名应为：lalala_k。但显然存在语法问题。我该怎么办？ THX

Answer 1

这只是一个普通的字符串;你可以用连接或字符串函数来构建它。例如：

col = var + '_k'

或

col = '{}_k'.format(var)