numpy.unravel_index() function | Python

numpy.unravel_index() function converts a flat index or array of flat indices into a tuple of coordinate arrays.

Syntax : numpy.unravel_index(indices, shape, order = ‘C’)
Parameters :
indices : [array_like] An integer array whose elements are indices into the flattened version of an array of dimensions shape.
shape : [tuple of ints] The shape of the array to use for unraveling indices.
order : [{‘C’, ‘F’}, optional] Determines whether the multi-index should be viewed as indexing in row-major (C-style) or column-major (Fortran-style) order.

Return : [tuple of ndarray] Each array in the tuple has the same shape as the indices array.

Code #1 :

Output :

(array([3, 6, 6]), array([4, 5, 1]))

 
Code #2 :

Output :

(array([1, 6, 2]), array([3, 5, 5]))