python初始化二维数组的小问题

 

python二维数组初始化问题

引子：今天在用Python写01背包的算法课作业时，发现一个关于二维数组的小问题，本来期末已经很忙了，但这个问题还是值得记录下来的，因为这不是第一次遇到了，这次花了将近一个小时才搞明白。

问题：

以如下方式初始化一个二维数组，在使用的过程中有没有问题？

matrix = [[0] * n] * m

有问题！可以做个实验：

>>> m = 3
>>> n = 4
>>> matrix = [[0] * n] * m
>>> matrix
[[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]
>>> matrix[1][2] = 1
>>> matrix
[[0, 0, 1, 0], [0, 0, 1, 0], [0, 0, 1, 0]]
>>>

可以看出，我只修改了matrix[1][2]这一个元素，但从结果来看，matrix这个二维数组的每一个一维数组的第3个（下标为2）元素都被修改了。对此我的理解是：在申请matrix数组时，“[0] * n”这个一维数组可能只申请了一个，“ * m”这操作，只是创建了m个指向“[0] * n”的引用，因此，若修改这个一维数组中的元素，则matrix中m个一位数组都会被修改。 更详细解释可以参考此篇博客在python中定义二维数组。

解决方案：

换种初始化的方式：

>>> matrix = [[0] * n for i in range(0, m)]
>>> matrix
[[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]
>>> matrix[1][2] = 1
>>> matrix
[[0, 0, 0, 0], [0, 0, 1, 0], [0, 0, 0, 0]]
>>>

可以看出，上述那种我们想要避免的情况已经不再出现了。

参考文献

1.在python中定义二维数组，woshare，博客园，http://www.cnblogs.com/woshare/p/5823303.html。