Python展平二维列表的行

Question

目前正在开发一个需要展平二维列表的列、行和对角线的程序。 我编写了以下代码来展平列和对角线，但无法展平行。 我不确定我执行不正确。

cols = []
rows = []
max_col = len(grid[0])
max_row = len(grid)

for y in range(max_row):
   rows.append(grid[y][x])
   for x in range(max_col):
       cols.append(grid[y][x])
print(cols)
print(rows)

firstDiagonal = [grid[i][i] for i in range(len(grid))]
secondDiagonal = [grid[i][len(grid)-1-i] for i in range(len(grid))]

输出类似于： ['G', 'A', 'O', 'C', 'T']['T', 'E', 'B', 'R', 'S'] 行输出应该不再模仿列输出。 ['A'，'S'，'T'，'R'，'V'，'Y'，'V'，'B'，'B'，'G']

Answer 1

要沿相反的维度展平，您需要交换循环的方向。 如果你通过做展平列

cols = []
for y in range(max_row):
    for x in range(max_col):
        cols.append(grid[y][x])

然后你做

rows = []
for x in range(max_col):
    for y in range(max_row):
        rows.append(grid[y][x])

您可以使用extend方法大大简化cols的计算，因为内部循环遍历整个行：

cols = []
for y in range(max_row):
    cols.extend(grid[y])

或者更好：

cols = []
for row in grid:
    cols.extend(row)

这样的简化不适用于rows ，但您可以使用zip使用 transpose 习惯用法：

rows = []
for col in zip(*grid):
    rows.extend(col)

最后，可以使用以下任何嵌套推导将每个表达式写成单行：

cols = [grid[y][x] for y in range(max_row) for x in range(max_col)]
cols = [item for row in grid for item in row]
rows = [grid[y][x] for x in range(max_col) for y in range(max_row)]
rows = [item for col in zip(*grid) for item in col]