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)