numpy.unravel_index

numpy.unravel_index(indices, dims, order=‘C’)
Converts a flat index or array of flat indices into a tuple of coordinate arrays.
意思是返回一个flatten矩阵的元素的index的横竖索引(x,y)。

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