python程序从用户输入生成n*n矩阵

Question

我想通过用户输入 n 的值和矩阵元素生成一个 n*n 矩阵。

代码如下:

n=int(input("Enter the matrix size"))

import numpy as np

#initialise nxn matrix with zeroes
mat=np.zeros((n,n))

#input each row at a time,with each element separated by a space
for i in range(n):
    for j in range(n):
        mat[i][j]=input()
print(mat)  

但是我得到这样的输出

[[1. 2.]

 [3. 4.]]

与一个。 (点)在我不想要的数字之后。有什么方法可以通过仅使用循环和数组而不是 NumPy 来实现吗？

Answer 1

你差一点。您只需在将矩阵初始化为

时将数据类型声明为 dtype=int

mat=np.zeros((n,n), dtype=int)

然后你就没有点了，只有

[[1 2]
[3 4]]