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)