获取 numpy 中另一个列表中元素的索引
Getting indices of elements that are in another list in numpy
我有两个 numpy 数组，我想获取第二个数组中第一个数组中所有元素的索引。例如：
import numpy as np
x = np.array([0,1,1,2,3,4,5,5])
y = np.array([1,3])
# want to get np.array([1,2,4])
如果 y 是一个标量，我可以只做 np.where(x == y)。是否有值数组的等价物？
你可以numpy.where with numpy.in1d:
>>> np.where(np.in1d(x, y))
(array([1, 2, 4]),)
我有两个 numpy 数组，我想获取第二个数组中第一个数组中所有元素的索引。例如：
import numpy as np
x = np.array([0,1,1,2,3,4,5,5])
y = np.array([1,3])
# want to get np.array([1,2,4])
如果 y 是一个标量，我可以只做 np.where(x == y)。是否有值数组的等价物？
你可以numpy.where with numpy.in1d:
>>> np.where(np.in1d(x, y))
(array([1, 2, 4]),)