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

Question

我想通过用户输入矩阵的值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？

标签： pythonpython-3.x

Answer 1

你几乎接近了。您只需将数据类型声明为dtype=int同时将矩阵初始化为

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

然后你不会有点，而只是

[[1 2]
[3 4]]