Matlab sub2ind到python

Question

row = [1 2 3 1];
col = [2 2 2 3];
sz = [3 3];
ind = sub2ind(sz,row,col)

Result:

ind = 1×4

     4     5     6     7

下面是MATLAB文档中的例子。我想在Python中用相同的结果生成相同的代码。我在其他文章中看到了三维数组如何为numpy数组(sub2ind)获得线性索引的例子，但我想要一个简单的二维数组的例子，如我所示。感谢回答代码！

Answer 1

def sub2ind(sz, row, col):
    n_rows = sz[0]
    return [n_rows * (c-1) + r for r, c in zip(row, col)]

首先获取行数。那么每个索引都是times（列-1）+行。

使用

>>> row = [1, 2, 3, 1];
>>> col = [2, 2, 2, 3];
>>> sz = [3, 3];
>>> sub2ind(sz, row, col)
[4, 5, 6, 7]