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

Question

我有一个看起来像这样的数据集：

id AttA AttB AttC
1   Y         Y
2        Y    

我想创建另一列，其中包含每种情况的属性总数，如下所示：

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

对我来说，应该如何解决这个问题并不明显，因为我对熊猫还很陌生。

预先感谢

Answer 1

您可以使用ne('')检查数据框中的哪些单元格不为空，然后将sum的设置axis设置为1：

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

   AttA AttB AttC  TotalAtts
0    Y         Y          2
1         Y               1

Answer 2

如果要计数Y，则可以执行(df == 'Y').sum(1)。如果您要计算非null值，则可以执行df.count(1)，但是此操作将计算空字符串。