python 链表推导式

链表推导式无非是为了生成链表,然后返回该链表值为前面所使用。

在写dp题目的时候,总是需要使用如下方式来生成初始矩阵:

n = 10
# 生成n个0
matrix = [0 for i in range(n)]
print(matrix)

等效于:

matrix = []
for i in range(n):
    matrix.append(0)
print(matrix)

结果:

[0, 0, 0, 0, 0, 0, 0, 0, 0, 0]

若要生成二维矩阵,可以这么写:

T = 15
n = 5
# 生成 n+1个 T*[0]+[1]的矩阵
matrix = [[1] + [0] * T for i in range(n+1)]
print(matrix)

结果:

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

当然也可以作为其它使用,以后用到再补充。

猜你喜欢

转载自blog.csdn.net/zdz0200/article/details/82684597