python - 如何使用列表理解从列表列表中的 "diagonal"范围创建列表？

Question

最终，我想使用reduce并找到一个产品。我知道如何使用 for 循环来执行这个问题，但我知道列表理解更快，而且通常更“Pythonic”，所以我尽最大努力理解如何在列表理解中进行我想要的移动。

这是一个例子:

grid = [[a, b, c, d]
        [e, f, g, h]
        [i, j, k, l]
        [m, n, o, p]
        [q, r, s, t]]  # All the sublists are integer lists.

所以最终我想做的是reduce()这些子列表的一些元素，以对角线方式选择它们。因此，在这个例子中，我想遍历长度为 3 的右下对角线。如果我从 [0][0] 开始，我的列表理解将通过 [a, f, k]。这是我迄今为止尝试过的一些代码:

diag_len = 3
i = 0
j = 0
# Purposely omitting the reduce() part of this next line
desired_list = grid[i:i + diag_len][j:j + diag_len]

实际上，上面的代码采用 grid 中索引 0、1 和 2 的整个子列表，而不仅仅是所需对角线中的三个元素。我当前的解决方案的 for 等效项如下所示:

for z in range(diag_len):
    desired_list.append(num_grid[i + z][j + z])
    desired_product *= desired_list[z]

我查看了 SO 上类似标记的问题，但找不到任何人使用列表理解同时跨越两个索引，如果我错过了，抱歉。

Answer 1

您可以使用列表理解沿着矩阵中的任何对角线创建项目列表，并输入行索引和列索引以及对角线的长度。然后使用 reduce() 获取结果列表中整数的乘积。

例如(您可能希望对函数进行更多保护，以防止输入值超出范围而导致错误，但这应该可以帮助您入门):

from functools import reduce

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

def get_diagonal(row, column, length):
    return [grid[row + i][column + i] for i in range(length)]

diag_list = get_diagonal(2, 1, 3)
diag_prod = reduce(lambda a, b: a * b, diag_list)

print(diag_list)
print(diag_prod)
# OUTPUT
# [10, 15, 20]
# 3000

如果您最终尝试组合成一个函数，您可以这样做:

def get_diagonal_product(row, column, length):
    return reduce(lambda a, b: a * b, [grid[row + i][column + i] for i in range(length)])