[FIXED] Adding new column as a sum of the subsquent columns

Issue

I have this df:

id   car  truck  bus  bike
0     1     1     0    0
1     0     0     1    0
2     1     1     1    1

I want to add another column count to this df but after id and before car to sum the values of the rows, like this:

id  count car  truck  bus  bike
0     2     1     1     0    0
1     1     0     0     1    0
2     4     1     1     1    1

I know how to add the column using this code:

df.loc[:,'count'] = df.sum(numeric_only=True, axis=1)

but the above code add the new column in the last position.

How can I fix this?

Solution

There are several ways. I provided two ways here.

#1. Changing column order after creating count column:

df.loc[:,'count'] = df.sum(numeric_only=True, axis=1)
df.columns = ['id', 'count', 'car', 'truck', 'bus', 'bike']
print(df)
#   id  count  car  truck  bus  bike
#0   0      2    1      1    0     0
#1   1      2    0      0    1     0
#2   2      6    1      1    1     1

#2. Inserting a Series to specific position using insert function:

df.insert(1, "count", df.sum(numeric_only=True, axis=1))
print(df)
#   id  count  car  truck  bus  bike
#0   0      2    1      1    0     0
#1   1      2    0      0    1     0
#2   2      6    1      1    1     1

Answered By - Park