python - 从数组中的一个 float 中查找值

Question

我有一个值数组，我想找到它在另一个值数组中的位置。例如，如果我有:

array1 = [1,2,3,4,5,6]
array2 = [2,6,3,4,1,5,.....]

我想找到数组 1 中的每个元素在数组 2 中的位置，所以我希望它返回类似的内容

what_position = [4,0,2,3,5,1]

我试过这样的:

for i in range(len(array1)):
      what_position = array1[i].index(array[2])

但是我得到一个错误提示

'numpy.float64' object has no attribute 'index'

我想这意味着我不能在 float 上使用 .index。有没有其他方法可以解决这个问题。

Answer 1

np.intersect1d 提供了另一种解决方案:

import numpy as np

array1 = [1,2,3,4,5,6]
array2 = [2,6,3,4,1,5]

np.intersect1d(array1, array2, return_indices=True)[2]