我想在数据框中生成一列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
我想避免迭代,所以是否有矢量化解决方案,如果是这样,它看起来怎么样?
答案 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