python - 如何填充二维数组？

Question

我有以下代码(几乎)填充Python中的列表列表(我将其称为二维数组)。

我希望它用数字 0 - 20 填充数组，而不是从 0-6 上升并重复 3 次。请参阅下面的代码:

#matrix=[1,2,3,4,5,6,7],[8,9,10,11,12,13,14],[15,16,17,18,19,20,21]
#print(matrix[1][2])

rows=3 
columns=7 

for i in range(rows):
  for j in range(columns):
    matrix=[j]
    i=i+1    
    print(matrix,end="")

错误的输出是:

[0][1][2][3][4][5][6][0][1][2][3][4][5][6][0][1][2][3][4][5][6]   

我希望输出是:

[1,2,3,4,5,6,7],[8,9,10,11,12,13,14],[15,16,17,18,19,20,21]

Answer 1

还有更奇特的方法，但这是最简单的:

>>> rows = 3
>>> columns = 7
>>> n = 1
>>> matrix = []
>>> for _ in range(rows):
...     sub = []
...     for _ in range(columns):
...         sub.append(n)
...         n += 1
...     matrix.append(sub)
...
>>> matrix
[[1, 2, 3, 4, 5, 6, 7], [8, 9, 10, 11, 12, 13, 14], [15, 16, 17, 18, 19, 20, 21]]

为了更好地衡量，一种奇特的方式:

>>> import itertools
>>> counter = itertools.count(1)
>>> rows = 3
>>> columns = 7
>>> matrix = [[n for n, _ in zip(counter, range(columns))] for _ in range(rows)]
>>> matrix
[[1, 2, 3, 4, 5, 6, 7], [9, 10, 11, 12, 13, 14, 15], [17, 18, 19, 20, 21, 22, 23]]
>>>