跟踪索引到 Python 中的原始矩阵

Question

我正在尝试追踪A2相对于A1的索引。 但是我得到一个错误。 随附所需的 output。

import numpy as np
A1 = np.array([[0.3, 1.2, 2],
     [3, 4, 5]]) # Shape 2 rows & 3 columns

A2 = np.array([0.3,1.2])
A=list(zip(*np.unravel_index(A2, np.array(A1).shape)))
print([A])

错误：

  File "<__array_function__ internals>", line 5, in unravel_index

TypeError: only int indices permitted

所需的 output：

[(0,0),(0,1)]

Answer 1

这将返回所需的元组列表：

A = [tuple(np.array(np.where(A1 == a)).ravel().tolist()) for a in A2]

对于A2中的每个元素a ，它使用np.where查找A1中的索引。 然后它使用ravel 、 tolist和tuple将索引转换为元组。 每个索引自动位于同一个列表中。