python矩阵列表转换索引计算

2021-03-28  本文已影响0人  xigua1234

首先,要知道数组有两种索引方式。比如对一个3x4的矩阵a(数据类型为:numpy.array)

unravel_index 已知线性索引,求矩阵索引

执行下面的代码

import numpy np
np.unravel_index([3, 0, 10], (3,4)) # 在一个(3x4)的矩阵中,第3、第0、第10个元素,的行列索引

结果是

(array([0, 0, 2], dtype=int64), array([3, 0, 2], dtype=int64))

解释:

对于一个(3x4)的矩阵,它的数据编号如下。所以输出结果就很好理解了

0 1 2 3
4 5 6 7
8 9 10 11

ravel_multi_index 已知矩阵索引,求线性索引

执行下面的代码。它是上面unravel_index例程的逆过程

import numpy np
np.ravel_multi_index (np.array([[0, 0, 2],[3, 0, 2]]),(3,4)) # 在一个(3x4)的矩阵中,求(0,3),(0,0),(2,2)位置处的线性索引

结果是

array([ 3,  0, 10], dtype=int64)
上一篇 下一篇

猜你喜欢

热点阅读