如何汇总熊猫中几列的所有类别变量的总和

Question

I have a dataset that looks something like this:

id AttA AttB AttC
1   Y         Y
2        Y    

I would like to create another column which has the total number of attributes for each case, as follows:

id AttA AttB AttC TotalAtts
1   Y         Y     2
2        Y          1

It's not obvious to me how I should approach this problem, since I'm fairly new to Pandas.

Thanks in advance

Answer 1

You could check which cells in the dataframe are not empty with ne('') , and take the sum setting axis to 1 :

df['TotalAtts'] = df.ne('').sum(1)

   AttA AttB AttC  TotalAtts
0    Y         Y          2
1         Y               1

Answer 2

If you want the count of Y , you can do (df == 'Y').sum(1) . If you want to count non-null values, then you can do df.count(1) , but empty strings will be counted by this.