链表推导式无非是为了生成链表,然后返回该链表值为前面所使用。
a = [[1, 2], [4, 5], [7, 8]]
b = [[] for i in a[0]]
print(b)
等价于:
x = []
for i in a[0]:
x.append([])
print(x)
结果:
[[], []]
1)在写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]
2)若要生成二维矩阵,可以这么写:
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]]
当然也可以作为其它使用,以后用到再补充。