numpy数组迭代,以跟踪目前为止的最大数量

Question

是否有一种简单的方法可以迭代二维数组来存储到目前为止在特定维度上遍历时获得的最大值.例如,我有一个数组:

[[2 ,  1,  5],
[-1, -1,  4],
[4, 3,  2],
[2 , 3, 4]]

我想要以下输出:

[[2 ,  2,  5],
[-1, -1,  4],
[4, 4,  4],
[2 , 3, 4]]

谢谢!

Answer 1

简单; 用途accumulate:

>>> numpy.maximum.accumulate(a, axis=1)
array([[ 2,  2,  5],
       [-1, -1,  4],
       [ 4,  4,  4],
       [ 2,  3,  4]])