我有一个零散布的一维数组。想创建第二个数组,其中包含最后一个零的位置,如下所示:
>>> a = np.array([1, 0, 3, 2, 0, 3, 5, 8, 0, 7, 12])
>>> foo(a)
[0, 1, 1, 1, 2, 2, 2, 2, 3, 3, 3]
是否有内置的 NumPy 函数或广播技巧可以在不使用 for 循环或其他迭代器的情况下执行此操作?
最佳答案
>>> (a == 0).cumsum()
array([0, 1, 1, 1, 2, 2, 2, 2, 3, 3, 3])
关于python - 一维 NumPy 数组中最后一个值的地址,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/39022027/