Python的二維數組操作

二維數組的初始化

最近刷題碰到需要用二維數組做的題目，所以就遇到了這樣的坑。

n = 3
m = 3
dp = [[0] * n] * m
dp[0][1] = 1
print dp
# output:
# [[0, 1, 0], [0, 1, 0], [0, 1, 0]]

上面這段代碼是想修改二維數組第1行第2列的數，可是居然三列都改了……

https://www.cnblogs.com/woshare/p/5823303.html
看了woshare的博客后，豁然開朗了

matrix = [array] * 3
……
也就是說matrix = [array] * 3操作中，只是創建3個指向array的引用，所以一旦array改變，matrix中3個list也會隨之改變。
那如何才能在python中創建一個二維數組呢？
例如創建一個3*3的數組
方法1 直接定義

[py]matrix = [[0, 0, 0], [0, 0, 0], [0, 0, 0]][/py]

方法2 間接定義

matrix = [[0 for i in range(3)] for i in range(3)]

修改一下代碼，舒服了

n = 3
m = 3
dp = [[0 for i in range(n)] for j in range(m)]
dp[0][1] = 1
print dp
# output:
# [[0, 1, 0], [0, 0, 0], [0, 0, 0]]

二維數組按列相加

n = 3
m = 3
like = [[1, 2, 3], [4, 5, 6], [0, 0, 0]]
res = [0 for i in range(n)]
for i in range(n):
    for j in range(m):
        res[i] += like[j][i] # 這里的i是列號
print res
# [5, 7, 9]

二維數組按行相加

n = 3
m = 3
like = [[1, 2, 3], [4, 5, 6], [0, 0, 0]]
res = [0 for i in range(n)]
for i in range(n):
    for j in range(m):
        res[i] += like[i][j] # 這里的i是行號
print res
# [6, 15, 0]