如何从用户那里获取矩阵输入并在 Python 中显示矩阵？

Question

我有 7 年的 Java 和 C++ 经验。我最近开始学习python。有人可以帮助我了解如何读取矩阵的输入并以矩阵格式显示输入。这是我写的代码：

 import sys

# no of rows are equal to the number of columns.
n = int(input("Enter the number of rows in a matrix"))
a = [[0 for x in range (n)] for y in range(n)]
for i in range (n):
    for j in range(n):
        a[i][j]=int(input())
        print (a[i][j])
    print("\n")

原文由 Shreya Srivastava 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

如果你使用熊猫，并制作一个数据框……如果你输入 4 作为行数，数字 1,2,3,4,5,6,7,8,9,10,11,12,13 ,14,15,16…下面的脚本

import pandas as pd
n = int(input("Enter the number of rows in a matrix: "))
a = [[0] * n for i in range(n)]
col_names = []
row_names = []

for i in range(n):
    col_names.append('col ' + str(i+1))
    row_names.append('row ' + str(i+1))
    for j in range(n):
          a[i][j] = int(input())

print(pd.DataFrame(a,columns = col_names, index = row_names))

…将产生以下输出。

        col 1  col 2  col 3  col 4
row 1      1      2      3      4
row 2      5      6      7      8
row 3      9     10     11     12
row 4     13     14     15     16

另一个选项…使用 numpy…和第一个选项中的相同值…

 import numpy as np
n = int(input("Enter the number of rows in a matrix: "))
a = [[0] * n for i in range(n)]

for i in range(n):
   for j in range(n):
        a[i][j] = int(input())

print(np.matrix(a))

…会产生

[[ 1  2  3  4]
 [ 5  6  7  8]
 [ 9 10 11 12]
 [13 14 15 16]]

原文由 Dave Rosenman 发布，翻译遵循 CC BY-SA 3.0 许可协议

Answer 2

    #program to print matirx by row and column input

a,b = map(int,input("Enter row and column of matrix separated by a space ").split())

# empty list for matrix
m=[]

# taking matrix input
for i in range(1,a+1):
    l=[]
    for j in range(1,b+1):
        n=int(input(f"Enter a[{i}][{j}] element "))
        l.append(n)

    m.append(l)

#printing in matrix form
for i in range(a):
    for j in range(b):
        if(j==b-1):
            print(m[i][j], end="")
        else:
            print(m[i][j], end=" ")
    if i!=a-1:
        print()

原文由 Aman 发布，翻译遵循 CC BY-SA 4.0 许可协议