Python:正确生成列表的方法

3,978 阅读2分钟

注意:在生成列表的时候,最好用 [0 for _ in range(n)] 的方式而不是 [0] * n 的方式生成,原因如下。

假设我们想要创建一个长度为 n = 2 列表 a 的时候,一般的做法有两种:a = [0] * na = [0 for _ in range(n)],在一维的时候两种方法没有区别。但是如果我们想要创建一个列表,列表中的每个元素都是长度为 m = 3 的列表时用两种方法出来的结果就是不同的:

n, m = 2, 3
a = [[0] * n] * m  # 用第一种方法生成的
# a = [[0, 0], [0, 0], [0, 0]]
b = [[0 for _ in range(n)] for _ in range(m)]  # 用第二种方法生成的
# b = [[0, 0], [0, 0], [0, 0]]

a[0][0] = 1  # 令 a 列表的第一个元素为 1
# a = [[1, 0], [1, 0], [1, 0]]
b[0][0] = 1
# b = [[1, 0], [0, 0], [0, 0]]

我们的预期结果是像 b 这样的,也就是第一个列表的第一个元素等于 1,但是列表 a 是将每一个列表的第一个元素都设为 1 了。

导致这种情况的原因在于,用第一种生成方法是类似于 = 的方式去生成的,也就是假设 a, b 都是列表,我们令 a = b ,如果他们其中一个的元素改变了,另一个也会跟着变,例如:

a = [0, 0, 0]
b = a
b[0] = 1
# a = [1, 0, 0]
# b = [1, 0, 0]

要想避免这样的情况,就应该这样写 b = list(a).

回到一开始的问题,如果用第一种方法 a = [[0] * n] * m,那么 a 中的 m 个列表存储的地址都是一样的,那么你改变其中的一个列表,其他的列表都会跟着改变,就会出现上面的情况。

c = [[0] * n for _ in range(m)]  # 没问题
d = [[0 for _ in range(n)]] * m  # 有问题

所以,以后要生成的列表时候,一律用 [0 for _ in range(n)] 会更好。