Python中向量化矩阵的对应坐标

Question

我已经回答了here 如何使用 Matlab 获得矢量化矩阵的相应坐标。

我想知道如何将答案更改为 Python。 Python 是否应用了存储数组元素的相同概念，但不是从 1 开始，而是从 0 开始？

通过应用与答案中提到的相同的数组元素存储，并通过在 Python 中手动执行 sub2ind，我正在做：

row = np.mod(i,m)
col = np.floor((i)/m)

我说的对吗？寻找可以为我提供更多详细信息的人，以防我错了。

Answer 1

Numpy 数组以行优先顺序存储，因此您需要

col = np.mod(i,m)
row = np.floor((i)/m)

可以更简单的写成

col = i%m
row = i//m

但是，numpy 具有 numpy.unravel_index 函数，无需手动计算即可完成此操作。