python - 宽数据的摘要统计

Question

是否有一种有效的方法来计算每个水果行为 True 的汇总统计数据？

df   comment  type      score    apple   banana   pear   
0     dfsd    new        0.4     True    False    True     
1     sdfs    low        0.3     False   True     False 
2     sdddfs   low       0.2     False   True     False    
3     sdsfs    low       0.8     True    True     False    
4     ddds    low        0.1     True    True     True

... 

我试过:

fruits = ['apple','banana','pear']

for fruit in fruits:
    df1 = df.loc[df.f'{fruit}', :]
    df1.describe()

预期输出:

fruit
        count     mean_score   std_score  
apple               
banana              
pear                

Answer 1

选择所需的 fruits 列，然后为每个 fruit 列获取相应的 score 并屏蔽 False 值，最后使用 describe 获取描述性统计数据:

s = ['count', 'mean', 'std']
stats = df[fruits].apply(lambda m: df['score'].mask(~m)).describe().T[s]

---

print(stats)

        count      mean       std
apple     3.0  0.433333  0.351188
banana    4.0  0.350000  0.310913
pear      2.0  0.250000  0.212132