gpt4 book ai didi

python - 一维 NumPy 数组中最后一个值的地址

转载 作者:太空宇宙 更新时间:2023-11-03 12:56:40 26 4
gpt4 key购买 nike

我有一个零散布的一维数组。想创建第二个数组,其中包含最后一个零的位置,如下所示:

>>> 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/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com