numpy.unravel_index的作用和解释

函数名：numpy.unravel_index(indices, dims)

功能：返回indices中的元素在维度为dims的数组中的索引值，默认按元组的形式返回。

举例：

1）numpy.unravel_index(indices=[22, 41, 37], dims=(7,6))，其返回元组(array([3, 6, 6]), array([4, 5, 1]))。

其过程类似：首先构造一个数组arr1=np.array(range(7*6)).reshape((7,6));返回indices中的元素值在数组arr1中对应值的索引位置。[3, 6, 6]分别对应[22, 41, 37]在第1维中的位置，[4, 5, 1]分别对应[22, 41, 37]在2维中的位置。

2）numpy.unravel_index(indices=1621, dims=(6,7,8,9))，其返回元组(3, 1, 4, 1)。

其过程类似：首先构造一个数组arr2=np.array(range(6*7*8*9)).reshape((6,7,8,9));返回indices中的元素值在数组arr2中对应值的索引位置。(3, 1, 4, 1)分别对应1621在第1维、第2维、第3维和第4维中的位置。

说明：函数numpy.unravel_index(indices, dims)返回的索引值从0开始计数。
原文：https://blog.csdn.net/qq_39366624/article/details/79918937