python - 如何在Python中创建一个矩形矩阵？

Question

m,n=input()
a=[0]*n
for i in range(0,m):
      a[i]=[0]*m
for i in range(0,m):
     for j in range(0,n):
           a[i][j]=input()
print a

考虑上面用 Python 2.7.4 编写的代码，接受一个二维数组，然后打印它。这段代码运行良好，但它应该接受任何二维数组，例如 m 和 的值可以分别为 3,2 ，但它只接受方阵。我们无法创建矩形矩阵，因为它会给出错误:如果 m 和 n 的值不相等，则索引超出范围。有没有办法像我们在 C/C++ 中一样轻松创建矩形矩阵？

Answer 1

Numpy 是一个用于快速线性代数运算的出色模块。您可以使用 Numpy 创建一个矩形数组，它本质上是一个矩阵。 (Numpy 也有矩阵函数，但使用起来有点乏味)。

作为示例，创建一个 3x4 数组如下

import numpy as np

input = np.zeros((3, 4))  #This creates a 3x4 array. It is good practice to initialize your array with 0's

input[0][3] = 5           #Fill out your array. The 0,0 index is the top left corner

In [42]:input
Out[42]:
array([[ 0.,  0.,  0.,  5.],
       [ 0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.]])

继续按照正常方式填写矩形矩阵的其余部分。