python的二维数组操作_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列的数，可是居然三列都改了……

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]