14
我正在寻找一个numpy函数来查找某个值在矢量(xs)内找到的索引。这些值在另一个数组(ys)中给出。返回的索引必须遵循ys的顺序。在Numpy数组中查找多个值
在代码中,我想用numpy函数替换下面的列表理解。
>> import numpy as np
>> xs = np.asarray([45, 67, 32, 52, 94, 64, 21])
>> ys = np.asarray([67, 94])
>> ndx = np.asarray([np.nonzero(xs == y)[0][0] for y in ys]) # <---- This line
>> print(ndx)
[1 4]
有没有快速的方法?
感谢
'ys'会很长吗? – kennytm 2012-03-05 12:27:19