numpy.unravel_index

numpy.unravel_index(indices, dims, order=’C’)
作用:将平面索引或平面索引数组转换为坐标数组。

  • Parameters:
    indices : array_like
    dims : 整型元组,指定原多维数组的维度
    order : {‘C’, ‘F’}, 可选,默认为C,决定索引是否看作C行主序,即坐标表示为(x, y)形式,数组是row在前,col在后。或者F列主序。

例子:

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.arange(42).reshape(7,6)
array([[ 0,  1,  2,  3,  4,  5],
       [ 6,  7,  8,  9, 10, 11],
       [12, 13, 14, 15, 16, 17],
       [18, 19, 20, 21, 22, 23],
       [24, 25, 26, 27, 28, 29],
       [30, 31, 32, 33, 34, 35],
       [36, 37, 38, 39, 40, 41]])
  • 可以对比一下就知道,比如22在这个矩阵下的坐标是(3,4).

你可能感兴趣的:(numpy)