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