Python：正确生成列表的方法

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

假设我们想要创建一个长度为 ​​n = 2​​​ 列表 ​​a​​​ 的时候，一般的做法有两种：​​a = [0] * n​​​ 和 ​​a = [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)]​​ 会更好。

以上就是本次分享的全部内容，现在想要学习编程的小伙伴欢迎关注Python技术大本营，获取更多技能与教程。