从用户输入生成 mxn 矩阵

Question

我是 python 新手，我正在尝试通过一些练习来介绍 numpy。我被这个问题困住了：

创建一个函数，它需要 ????,???? ∈ ℕ 作为输入并生成一个????×????矩阵（numpy.array）？？？？带有条目a[i,j] = j*m + i，其中 0 ≤ ???? ≤ ????−1 和 0 ≤ ???? ≤????−1

我已经找到了一种或多或少无需 numpy 的方法，但我们将不胜感激。

Answer 1

我认为最好的方法是先使用生成器创建一个数组，然后将其转换为一个 numpy 数组

// m, n = rows, cols
np.array([j*m + i for j in range(n) for i in range(m)]).reshape((m, n))

但是，它是沿行的数字序列，因此可以更轻松地完成

np.array(range(m*n)).reshape((m, n))

不过，对于这个 numpy 库，还有一个内置函数 arange

np.arange(m*n).reshape((m, n))

希望对你有帮助。