Python numpy unravel_index用法及代码示例

本文简要介绍 python 语言中 numpy.unravel_index 的用法。

用法:

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

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

参数：

indices： array_like

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

shape： 整数元组

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

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

确定索引是否应被视为按行优先(C 风格)或列优先(Fortran-style)顺序进行索引。

返回：

unraveled_coords： ndarray 的元组

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

例子：

>>> 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)