python - 如何构建5x5矩阵？

Question

如何构造一个 5 行 5 列的矩阵？

lst = [1,2,3,4,5]

[[float("inf")]*len(lst) for k in range (len(lst))]

给我[[inf, inf, inf, inf, inf], [inf, inf, inf, inf, inf], [inf, inf, inf, inf, inf], [inf, inf, inf，inf，inf]，[inf，inf，inf，inf，inf]]

如何更改参数才能获得 5x5 矩阵？

Answer 1

我不确定您为什么使用 lst 变量，但您需要的是接近此的东西:

def matrix(x,y,initial):
    return [[initial for i in range(x)] for j in range(y)]

这给出:

> print matrix(5,5,float('inf'))
[[inf, inf, inf, inf, inf], [inf, inf, inf, inf, inf], [inf, inf, inf, inf, inf], [inf, inf, inf, inf, inf], [inf, inf, inf, inf, inf]]

> my_matrix = matrix(2,2,0)
> print my_matrix
[[0, 0], [0, 0]]
> my_matrix[0][2] = 2
> print my_matrix
[[0, 2], [0, 0]]

大多数语言中的矩阵只是一组嵌套数组。如果您需要更多的东西，您可能需要创建一个自定义类。