Numpy:快速找到值的第一个索引 [英] Numpy: find first index of value fast

问题描述

如何在 Numpy 数组中找到数字第一次出现的索引?速度对我很重要.我对以下答案不感兴趣，因为它们会扫描整个数组并且在找到第一次出现时不会停止:

How can I find the index of the first occurrence of a number in a Numpy array? Speed is important to me. I am not interested in the following answers because they scan the whole array and don't stop when they find the first occurrence:

itemindex = numpy.where(array==item)[0][0]
nonzero(array == item)[0][0]

注 1:该问题的所有答案似乎都不相关 是否有一个 Numpy 函数可以返回数组中某事物的第一个索引?

Note 1: none of the answers from that question seem relevant Is there a Numpy function to return the first index of something in an array?

注意 2:使用 C 编译的方法优于 Python 循环.

Note 2: using a C-compiled method is preferred to a Python loop.

推荐答案

Numpy 2.0.0 有一个功能请求:https://github.com/numpy/numpy/issues/2269

There is a feature request for this scheduled for Numpy 2.0.0: https://github.com/numpy/numpy/issues/2269

这篇关于Numpy:快速找到值的第一个索引的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！