在 Python 中遍历列表列表中的列

Question

当我尝试跨行中的列进行迭代时，该列在嵌套循环中没有变化：

i_rows = 4
i_cols = 3
matrix = [[0 for c in xrange(i_cols)] for r in xrange(i_rows)]

for row, r in enumerate(matrix):
    for col, c in enumerate(r):
        r[c] = 1

print matrix

观察到的输出

[[1, 0, 0], [1, 0, 0], [1, 0, 0], [1, 0, 0]]

预期输出

[[1, 1, 1], [1, 1, 1], [1, 1, 1], [1, 1, 1]]

我尝试了不同的表达方式，例如 xrange() 和 len()，我正在考虑切换到 numpy。我有点惊讶的是，Python 中的二维数组并没有我对这门语言的第一印象那么直观。

目标是一个具有不同整数值的二维数组，稍后我需要对其进行解析以表示屏幕上的 2D 图形。

如何遍历列表列表中的列？

Answer 1

您只需将值分配给col，而不是c

for row, r in enumerate(matrix):
    for col, c in enumerate(r):
        r[col] = 1               # Note `col`, not `c`

因为enumerate 返回的第一个值将是索引，而第二个值将是实际值本身。