如何用更优雅方式实现Numpy数组的前向最大值填充？

你的需求本质是计算数组的累积最大值——从左到右遍历数组时，每一位保留到当前位置为止的最大元素，这样自然就把所有比前一个元素小的值替换成了之前的更大元素。

用Numpy内置的np.maximum.accumulate()可以一步完成，完全不需要循环，代码更简洁高效：

import numpy as np

def func_elegant(a):
    return np.maximum.accumulate(a)

测试你的示例：

# 第一个示例
arr1 = np.array([10, -1, 2, 5, 19, 5, 5, 4, 10, 2])
print(func_elegant(arr1))
# 输出: [10 10 10 10 19 19 19 19 19 19]

# 第二个示例
arr2 = np.array([0, 3, 5, 4, 3, 7, 2])
print(func_elegant(arr2))
# 输出: [0 3 5 5 5 7 7]

方案优势 #

• 矢量化优化：Numpy内置函数是底层优化实现的，比Python手动循环快得多，处理大数组时性能差距尤为明显。
• 代码简洁：一行核心代码完成逻辑，可读性更强，无需手动处理循环索引和元素替换。
• 无副作用：np.maximum.accumulate()返回新数组，不会修改原输入，和你原代码的copy()行为保持一致。

如果需要原地修改数组以节省内存，也可以这样写：

def func_inplace(a):
    np.maximum.accumulate(a, out=a)
    return a

内容的提问来源于stack exchange，提问作者user3306199