Calculate sum of specific rows using Python

Question

I need to calculate the sum of specific rows in my dataframe

for example, Nombre de reboot, Passage en mode privé, Passage en mode public, Nombre de Kilomètres parcourus, Heures de roulage,Temps de trajet..

I tested this code on the first three rows:

import pandas as pd
df = pd.read_excel('mycollected_data1.xlsx')


print (df.iloc[:3, df.columns.get_indexer(['Valeurs','Valeurs.1','Valeurs.2'])])
a = df.iloc[1:3, [1,2,3]].sum()

print ('a\n', a)

this is the output:

    Valeurs Valeurs.1 Valeurs.2
0       3         5         0
1       2         1         1
2       0         0         0
a
 Valeurs      2.0
Valeurs.1    1.0
Valeurs.2    1.0
dtype: float64

desired output:

        Valeurs Valeurs.1 Valeurs.2   sum
    0       3         5         0       8
    1       2         1         1       4
    2       0         0         0       0

how can I make it calculate the sum of specific given rows?

quant · Accepted Answer · 2019-11-25 13:39:10Z

2

You need to use axis=1 in your sum function

Original df

df
   Valeurs  Valeurs.1  Valeurs.2
0        3          5          0
1        2          1          1
2        0          0          0


cols = ['Valeurs','Valeurs.1','Valeurs.2']

df['sum'] = df.loc[0:2, cols].sum(axis=1)
df # final df

   Valeurs  Valeurs.1  Valeurs.2  sum
0        3          5          0    8
1        2          1          1    4
2        0          0          0    0

answered Nov 25, 2019 at 13:39

quant

4,5125 gold badges38 silver badges84 bronze badges

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

Comments

Umar.H · Accepted Answer · 2019-11-25 13:49:24Z

2

IIUC,

we can use .sum while specifying the axis as 1 to work row wise.

cols = ['Valeurs','Valeurs.1','Valeurs.2']

df['sum'] = df[cols].sum(axis=1)

print(df)

   Valeurs  Valeurs.1  Valeurs.2  sum
0        3          5          0    8
1        2          1          1    4
2        0          0          0    0

edit, if you need to access certain rows you can use the .loc function as @quant has

row_start = 0
row_end = 2
df.loc[row_start:row_end,cols].sum(axis=1)

Just as a word of advice, it seems like you have repeated column names with similar datatypes, i would first clean your column headings and then melt your dataframe to get a tabular model.

edited Nov 25, 2019 at 13:49

answered Nov 25, 2019 at 13:25

Umar.H

23.1k7 gold badges50 silver badges94 bronze badges

2 Comments

newbie Over a year ago

thank you, but how can I specify the row that I need to work on ?

Umar.H Over a year ago

as in you only want to sum a certain index location? will the other values just be NaN?if so you can use loc.

Collectives™ on Stack Overflow

Calculate sum of specific rows using Python

2 Answers 2

Comments

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related