python - 用numpy数组中第一个出现的非零值替换所有零值

Question

我有一个 numpy 数组，看起来像：

[0,0,0,1,2,0,0,3,0,0,9]

我想将此数组转换为：

[1,1,1,1,2,3,3,3,9,9,9]

如您所见，所有零值都已替换为下一个非零值。有没有办法numpy可以做到这一点？

标签： pythonnumpy