跟踪数据框列

时间:2016-06-09 08:07:34

标签: python pandas

我想在数据框中生成一列B,该列跟踪自索引0行以来列A中达到的最大值。

          A     B
 Index    
   0      1     1
   1      2     2
   2      3     3
   3      2     3
   4      1     3
   5      3     3
   6      4     4
   7      2     4

我想避免迭代,所以是否有矢量化解决方案,如果是这样,它看起来怎么样?

1 个答案:

答案 0 :(得分:4)

您正在寻找cummax

In [257]:
df['B'] = df['A'].cummax()
df

Out[257]:
       A  B
Index      
0      1  1
1      2  2
2      3  3
3      2  3
4      1  3
5      3  3
6      4  4
7      2  4