使用python pandas计算概率

时间:2017-04-28 00:25:58

标签: python pandas

我想计算飞机类型的概率。为简单起见;

Airplane Accident
A           1    
A           1
A           1
A           0
B           1
B           0
B           0
C           1
C           0

我想得到的是第一个

Airplane   prob
A          0.75
B          0.33
C          0.5

最后想获得0.75的最大值。怎么做?

1 个答案:

答案 0 :(得分:4)

df.groupby('Airplane')['Accident'].mean()

输出:

Airplane
A    0.750000
B    0.333333
C    0.500000

然后获得最大值:

df.groupby('Airplane')['Accident'].mean().max()

输出:

0.75