数组1中每个元素X的Numpy查找数组2中第一个大于X的元素的索引

2024-06-02 11:18:36 发布

您现在位置:Python中文网/ 问答频道 /正文

我有两个numpy数组xy

x = numpy.array([1,2,3,4,5])
y = numpy.array([1,3,5,7])

z = func(x,y) # z = [2,2,3,3,4]

是否有一个函数func将返回上述内容。例如,y中第一个大于1的元素(x中第一个元素)的索引是2,因此z[0]=2

寻找一个线性向量化的方法来做这件事,如果它存在的话。对实际比较函数不可知的东西(小于、大于、leq、geq等)


Tags: 方法函数numpy元素内容线性数组array