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)