1d NumPy数组中最后一个值的地址

时间:2016-08-18 15:24:59

标签: python arrays numpy

我有一个1d数组,零点分散在各处。想创建一个包含最后一个零位的第二个数组,如下所示:

>>> 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循环或其他迭代器的情况下执行此操作?

1 个答案:

答案 0 :(得分:8)

void foo(int *pList)
{
    foo(pList); // recursive
}