0%

Python 中生成包含空列表的特定长度的列表

最近用 Python 的时候,踩了一个坑。

主需求,是将横向排列为主的二维数组,读入序列化为纵向排列为主的二维数组。为此,需要先确定原数据的列数,然后建立一个特定长度的空列表:

1
[[], [], [], ... [], []]

这是个坑

Python 的乘法非常神奇。比如有:

1
2
3
>>> a = [[]] * 2
>>> a
[[], []]

这个方式非常优雅,所以我首先想到的,就是用这样的乘法生成符合要求的二维空数组。

trap.py
1
2
3
4
5
6
width = 3
a = [list()] * 3
b = range (width + 1)
for i in xrange (width):
a[i].append (b[i + 1])
print a

预想中,最后的 print a 应该输出 [[1], [2], [3]],然而实际上却输出了

1
[[1, 2, 3], [1, 2, 3], [1, 2, 3]]

看到这个结果,我直接懵逼了。

出坑

原来,Python 里的乘法创建的是实例的引用。也就是说,[list()] * 3 里的(作为元素的)列表,实际是一个实例:修改其中之一,会同步修改到剩余所有的引用。也就是有:

1
2
3
4
5
6
>>> a = [[1]] * 2
>>> a
[[1], [1]]
>>> a[0][0] = 2
>>> a
[[2], [2]]

因此,为了达到我最开始的目的,不能用「魔术乘法」,只能老实地用列表推导式。如:

answer.py
1
2
3
4
5
6
width = 3
a = [list() for i in xrange (width)]
b = range (width + 1)
for i in xrange (width):
a[i].append (b[i + 1])
print a

得到预期:

1
[[1], [2], [3]]
俗话说,投资效率是最好的投资。 如果您感觉我的文章质量不错,读后收获很大,预计能为您提高 10% 的工作效率,不妨小额捐助我一下,让我有动力继续写出更多好文章。