Sum pandas dataframe rows using column starting with same name

Question

I have this DataFrame:

import pandas as pd
d = {'1_col': [1, 2], '2_col': [3, 4], 'var1': [5,6]}
df = pd.DataFrame(data=d)
print(df)

which looks like this:

    1_col  2_col  var1
0     1     3     5
1     2     4     6

I need to sum all the columns whose name ends in _col, so that the resulting DataFrame looks like this (the column called sum is the sum of the columns 1_col and 2_col:

    1_col  2_col  var1 sum
0     1     3     5     4
1     2     4     6     6

Is there a way in pandas to sum all the columns whose name ends with "_col" rather than doing it manually?

Mayank Porwal · Accepted Answer · 2022-02-16 13:31:16Z

3

Use df.filter with df.sum:

In [1209]: df['sum'] = df.filter(like='_col').sum(1)

In [1210]: df
Out[1210]: 
   1_col  2_col  var1  sum
0      1      3     5    4
1      2      4     6    6

answered Feb 16, 2022 at 13:31

Mayank Porwal

34.2k9 gold badges45 silver badges65 bronze badges

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

Collectives™ on Stack Overflow

Sum pandas dataframe rows using column starting with same name

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