时间戳

首页 > 解决方案 > 在numpy数组中传播最大值

python - 在numpy数组中传播最大值

问题描述

我认为这是一个相对简单的问题，因为我知道 numpy 中有一个简单的答案，但我再也找不到了。

我有一个 numpy 数组，比如说[1, 2, 10, 5, 6, 12, 9, 9, 8]

我想从头到尾遍历它并只保留迄今为止看到的最大值，例如答案是[1, 2, 10, 10, 10, 12, 12, 12, 12]

用一个循环很容易解决，但前段时间我了解了一个 numpy 函数，它可以快速完成工作，而且也很紧凑。我正在寻找这种方法。

谢谢 ！

标签： pythonarrayspython-3.xnumpynumpy-ndarray

解决方案

您可以使用numpy.ufunc.accumulate, 来累积应用函数的结果，maximum在这种情况下，在所有元素上：

a = np.array([1, 2, 10, 5, 6, 12, 9, 9, 8])

np.maximum.accumulate(a)
# array([ 1,  2, 10, 10, 10, 12, 12, 12, 12], dtype=int32)

推荐阅读