使用熊猫进行数据框架统计分析

Question

我有一个数据框有几列。其中一列中的值为总和为1的百分比，例如列A.我想在另一列（列B）中打印这些值，以使这些值与列A中的最大值（总和为0.95）相对应。

例如：

Column A  Column B 
    A1    0.2 
    A2    0.5 
    A3    0.25 
    A4    0.01 
    A5    0.03 
    A6    0.01 

结果应该是

A1 
    A2 
    A3 

在列B总结这些值的相应值，以0.95

Answer 1

让我们用cumsum使用.loc布尔索引：

df.loc[df['Column B'].cumsum() <= .95,'Column A'] 

编辑：hausdork指出，如果我们想用最大的值，那么我们可以sort_values做到这一点：

df.loc[df.sort_values(by='Column B', ascending=False)['Column B'].cumsum() <= .95,'Column A'] 

输出：

0 A1 
1 A2 
2 A3 
Name: Column A, dtype: object