python - 跟踪数据框列中的最大值

Question

我想在数据框中生成一个 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

我想避免迭代，那么是否有矢量化的解决方案，如果有，它看起来像什么？

Answer 1

您正在寻找 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