问题描述
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 ++中一样轻松地创建矩形矩阵吗?
1楼
您可以执行以下操作:
rows = input()
cols = input()
arr = [[None] * cols for _ in range(rows)]
for i in range(rows):
for j in range(cols):
print(arr)
arr[i][j] = input()
print(arr)
您应该看到类似的问题:
至于使您的确切代码正常工作,您只需在第2行和第4行打开m
和n
:
m = input() # rows
n = input() # cols
a = [0] * m # Initialize rows
for i in range(0, m): # For each row (must be n rows)
a[i] = [0] * n # Initialize a column
for i in range(0, m): # For each row
for j in range(0, n): # For each column
a[i][j] = input() # Input a value
print a
2楼
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.]])
继续按照正常填充矩形矩阵的其余部分。