用 numpy.array 制作矩阵

Question

我尝试使用带有以下代码的 numpy.array 创建一个矩阵

def matrix_input(3):
    matrix = []
    for i in range(N):
        a = nd.array(input().split(),int)
        matrix.append(a)
    print(matrix)

但我得到以下输出： [array([1, 1, 1]), array([1, 1]), array([1, 1, 1])]

对于输入：

1 1 1 
 1 1 
1 1 1 

我不希望矩阵中包含单词数组...如何删除它？

Answer 1

将其列在代码的第 4 行。此外，请按照以下代码中的说明更正您的功能。 函数调用和函数创建是两件不同的事情，你传递给它的参数也是如此。

import numpy as np

def matrix_input(N):  # Argument to function while creation is wrong, use N instead of 3.
    matrix = []
    for i in range(N):
        a = list(np.array(input().split(),int))  # Make it a list here
        matrix.append(a)
    print(matrix)

输出：

matrix_input(3)
1 1 1
1 1
1 1 1

[[1, 1, 1], [1, 1], [1, 1, 1]]

创建正确矩阵的替代方法：

import numpy as np

matrix_1 = np.matrix([[1,1,1],[1,1,0],[1,1,1]])
print(matrix_1)

输出：

[[1 1 1]
 [1 1 0]
 [1 1 1]]