numpy.unravel_index

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

Преобразует плоский индекс или массив плоских индексов в кортеж массивов координат.

Параметры:

indices : array_like Массив целых чисел, элементы которого являются индексами в сплющенной версии массива с размерностью shape. До версии 1.6.0 эта функция принимала только одно значение индекса. shape : tuple of ints Форма массива, используемого для распаковки indices. Изменено в версии 1.16.0: Переименовано из dims в shape. order : {‘C’, ‘F’}, optional Определяет, должны ли индексы рассматриваться как индексы в порядке строк (стиль C) или столбцов (стиль Fortran). Добавлена в версии 1.6.0.

Возвращает:

unraveled_coords : tuple of ndarray Каждый массив в кортеже имеет ту же форму, что и массив indices.

См. также

ravel_multi_index

Примеры

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