当前位置: 首页>>编程语言>>正文


Python二维数组的初始化

这里说的数组其实是python中的list, 因为它可以按索引(下标)取元素值,所以相当于是一个arraylist。

我们知道,一位数组的简单初始化如下:

arr = [0] * 10  # 即[0, 0, 0, 0, 0, 0, 0, 0, 0, 0]

这样可以得到一个长度为10的数组。

那么二维情况下,能不能这么做呢?

比如 arr2D = [[0] * 10 ] * 4 能按预期得到一个正常的10×4二维数组吗?

答案是否定的。这样做的确非常简洁,但却是有问题的,原因说明如下:

由于[0]*10是一个数组对象, [[0]*10]*4相当于把对象的引用复制了4次,也就是说修改10×4数组的任何一个元素,其他引用位置的元素也会被改变!

例如下面的例子,这个结果显然不是我们想要的。

>> a = [[0]*10]* 4
>> print a
[ [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
  [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
  [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
  [0, 0, 0, 0, 0, 0, 0, 0, 0, 0]]
>> a[0][0] = 12345
>> print a
[ [12345, 0, 0, 0, 0, 0, 0, 0, 0, 0],
  [12345, 0, 0, 0, 0, 0, 0, 0, 0, 0],
  [12345, 0, 0, 0, 0, 0, 0, 0, 0, 0],
  [12345, 0, 0, 0, 0, 0, 0, 0, 0, 0]]

事实上,在python中,要初始化二位数组,一种比较简洁的正确方法是:


arr2D = [[ 0 for col in range(10) ] for row in range(4) ]

之前在实现某ML算法,存语料中的多个样本时踩了这个坑,所以记录下来加深下印象。

如果需要更高维度的数组,这种初始化方法就比较绕了,可以考虑将多维的元素表示成一个结构体,然后来个一维的数组就可以了。

本文由《纯净天空》出品。文章地址: https://vimsky.com/article/790.html,未经允许,请勿转载。