Python二维数组构造

周末用python要写个算法用到来二维数组, 一时间还不知道python怎么构造多维数组出来。看到一段不错的代码, 记录一下。

Python使用list嵌套实现多维数组, PHP可以使用array嵌套实现多维数组。

如果经常使用数组, 可以使用python的numpy包。

下面是个简单的例子:

 import sys 

 def create1D(length, value=None):
     """
     Create and return a 1D array containing length elements,
     each initialized to value.
     """
     return [value] * length

 def create2D(rowCount, colCount, value=None):
     """
     Create and return a 2D array having rowCount rows and
     colCount columns, with each element initialized to value.
     """
     a = [None] * rowCount
     for row in range(rowCount):
         a[row] = [value] * colCount
     return a

 if __name__ == '__main__':
     arr2D = create2D(3, 4, 0)
     arr2D[1][3] = 5
     for i in range(3):
         for j in range(4):
             sys.stdout.write(str(arr2D[i][j]) + " ")
         sys.stdout.write("\n")

执行结果:

0 0 0 0
0 0 0 5
0 0 0 0