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]