numpy.unravel_index#

numpy.unravel_index(indices, shape, order='C')#

将扁平索引或扁平索引数组转换为坐标数组元组。

参数:
indicesarray_like

一个整数数组,其元素是索引到维度为 shape 的数组的扁平化版本。在 1.6.0 版本之前,此函数只接受一个索引值。

shape整数元组

用于解开 indices 的数组的形状。

在 1.16.0 版本中变更: dims 重命名为 shape

order{‘C’, ‘F’}, 可选

确定索引是按行主序(C 样式)还是按列主序(Fortran 样式)进行查看。

在 1.6.0 版本中新增。

返回值:
unraveled_coordsndarray 元组

元组中的每个数组都与 indices 数组具有相同的形状。

示例

>>> import numpy as np
>>> np.unravel_index([22, 41, 37], (7,6))
(array([3, 6, 6]), array([4, 5, 1]))
>>> np.unravel_index([31, 41, 13], (7,6), order='F')
(array([3, 6, 6]), array([4, 5, 1]))
>>> np.unravel_index(1621, (6,7,8,9))
(3, 1, 4, 1)