Python function pass-in variable to be used as name inside function

Question

I'd like to change column name of dataframe:

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')

I would like second column name as var_'k'; that is, if I passed in lalala, the column name should be: lalala_k. But obviously there's syntax problem. What should I do? Thx

Daniel Roseman · Accepted Answer · 2017-04-09 16:33:09Z

4

This is just a normal string; you can build it up with concatenation or string functions. For example:

col = var + '_k'

or

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

answered Apr 9, 2017 at 16:33

Daniel Roseman

602k68 gold badges910 silver badges923 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Python function pass-in variable to be used as name inside function

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related