当前位置: 代码迷 >> python >> 如何在Python中创建矩形矩阵?
  详细解决方案

如何在Python中创建矩形矩阵?

热度:127   发布时间:2023-06-13 15:30:45.0
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 ++中一样轻松地创建矩形矩阵吗?

您可以执行以下操作:

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行打开mn

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

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.]])

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