Loop to change dataframe titles - python

Question

I am trying to loop through the titles of a dataframe and change them to the last character of the title so then I can convert them to months. I want a loop because I get data each month and I want to have it automated so I don't manually add the title names. When I print the titles this is what I get:

list(df)

['MCA036_LIFETIME_1', 'MCA036_LIFETIME_2', 'MCA036_LIFETIME_3', 'Total']

I would like for the titles to be:

list(df)

['January','February','March','Total']

Any guidance would be greatly appreciated.

llllllllll · Accepted Answer · 2018-05-17T18:56:48.580

3

You don't need loop, just:

df.columns = ['January','February','March','Total']

If you want to automatically convert the string in that list, it's a general Python problem, not particularly related to pandas:

In [141]: all_monthes = ['January', 'February', 'March']

In [142]: l = ['MCA036_LIFETIME_1', 'MCA036_LIFETIME_2', 'MCA036_LIFETIME_3', 'Total']

In [143]: [all_monthes[int(e.split('_')[-1])-1] if e != 'Total' else e for e in l]
Out[143]: ['January', 'February', 'March', 'Total']

edited May 17 '18 at 18:56

answered May 17 '18 at 18:47

llllllllll

16,169
4
31
54

The reason I want to loop through is so I can apply it when I collect more months of data. – Michael Ryan May 17 '18 at 18:48
@MichaelRyan See edit. It's just a list comprehension. – llllllllll May 17 '18 at 18:57

Loop to change dataframe titles - python

1 Answers1