我在numpy中有一个1D数组,我想找到索引的位置,其中一个值超过numpy数组中的值。
E.g.
aa = range(-10,10)
在aa中找到超过5的位置。
我在numpy中有一个1D数组,我想找到索引的位置,其中一个值超过numpy数组中的值。
E.g.
aa = range(-10,10)
在aa中找到超过5的位置。
当前回答
我会说
i = np.min(np.where(V >= x))
其中V是向量(1d数组),x是值,i是结果索引。
其他回答
给定数组的排序内容,还有一个更快的方法:searchsorted。
import time
N = 10000
aa = np.arange(-N,N)
%timeit np.searchsorted(aa, N/2)+1
%timeit np.argmax(aa>N/2)
%timeit np.where(aa>N/2)[0][0]
%timeit np.nonzero(aa>N/2)[0][0]
# Output
100000 loops, best of 3: 5.97 µs per loop
10000 loops, best of 3: 46.3 µs per loop
10000 loops, best of 3: 154 µs per loop
10000 loops, best of 3: 154 µs per loop
In [34]: a=np.arange(-10,10)
In [35]: a
Out[35]:
array([-10, -9, -8, -7, -6, -5, -4, -3, -2, -1, 0, 1, 2,
3, 4, 5, 6, 7, 8, 9])
In [36]: np.where(a>5)
Out[36]: (array([16, 17, 18, 19]),)
In [37]: np.where(a>5)[0][0]
Out[37]: 16
这样更快一些(看起来也更好一些)
np.argmax(aa>5)
因为argmax将在第一个True处停止(“如果最大值出现多次,则返回第一次出现的对应索引。”),并且不会保存另一个列表。
In [2]: N = 10000
In [3]: aa = np.arange(-N,N)
In [4]: timeit np.argmax(aa>N/2)
100000 loops, best of 3: 52.3 us per loop
In [5]: timeit np.where(aa>N/2)[0][0]
10000 loops, best of 3: 141 us per loop
In [6]: timeit np.nonzero(aa>N/2)[0][0]
10000 loops, best of 3: 142 us per loop
我对此也很感兴趣,我将所有建议的答案与perfplot进行了比较。(声明:我是perfplot的作者。)
如果您知道您正在查看的数组已经排序,那么
numpy.searchsorted(a, alpha)
是给你的。它是O(log(n))操作,也就是说,速度几乎不取决于数组的大小。没有比这更快的了。
如果你对数组一无所知,你就不会出错
numpy.argmax(a > alpha)
已经排序:
未分类的:
代码重现情节:
import numpy
import perfplot
alpha = 0.5
numpy.random.seed(0)
def argmax(data):
return numpy.argmax(data > alpha)
def where(data):
return numpy.where(data > alpha)[0][0]
def nonzero(data):
return numpy.nonzero(data > alpha)[0][0]
def searchsorted(data):
return numpy.searchsorted(data, alpha)
perfplot.save(
"out.png",
# setup=numpy.random.rand,
setup=lambda n: numpy.sort(numpy.random.rand(n)),
kernels=[argmax, where, nonzero, searchsorted],
n_range=[2 ** k for k in range(2, 23)],
xlabel="len(array)",
)
我会说
i = np.min(np.where(V >= x))
其中V是向量(1d数组),x是值,i是结果索引。