Arrays 从用户输入生成mxn矩阵

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

---

创建一个需要的函数我认为最好的方法是首先使用generator创建一个数组，然后将其转换为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))

希望能有帮助