使这个嵌套的 for 循环更加 Pythonic

Question

我有一段笨拙（但有效）的代码，如图所示：

plus_cords = []
for i in range(len(pluses)):
    plus_cords.append([ [pluses[i][0], pluses[i][1]] ])
    for j in range(1, pluses[i][2] + 1):
        plus_cords[i].append([pluses[i][0] - j, pluses[i][1]])
        plus_cords[i].append([pluses[i][0] + j, pluses[i][1]])
        plus_cords[i].append([pluses[i][0], pluses[i][1] - j])
        plus_cords[i].append([pluses[i][0], pluses[i][1] + j])

其中 'pluses' 是由 3 个整数组成的列表。

pluses = [[0, 0, 0], [0, 1, 0], [0, 2, 0], [0, 3, 0], [0, 4, 0], [0, 5, 0], [1, 0, 0], [1, 4, 0], [2, 0, 0], [2, 1, 0], [2, 2, 0], [2, 3, 0], [2, 4, 0], [2, 4, 1], [2, 5, 0], [3, 0, 0], [3, 1, 0], [3, 4, 0], [4, 0, 0], [4, 1, 0], [4, 2, 0], [4, 3, 0], [4, 4, 0], [4, 5, 0]]

我正在寻找有关如何使其更具可读性和效率，基本上更“pythonic”的想法。

提前谢谢你

Answer 1

与 Samwise 的答案类似，但分解为单独的步骤，实际上产生的结果与您的原始代码相同：

首先，当然，我们可以直接迭代pluses 的元素，而不是使用索引，然后将它们解包为它们的三个构成值，这已经使代码更具可读性：

plus_cords = []
for p0, p1, p2 in pluses:
    cords = [ [p0, p1] ]
    for j in range(1, p2 + 1):
        cords.append([p0 - j, p1])
        cords.append([p0 + j, p1])
        cords.append([p0, p1 - j])
        cords.append([p0, p1 + j])
    plus_cords.append(cords)

然后，我们可以尝试将内部循环转换为列表推导式。这有点棘手，因为我们必须使用嵌套生成器并将其解压缩到列表推导中以区分单个第一个元素和其余元素：

plus_cords = []
for p0, p1, p2 in pluses:
    cords = [[p0, p1], *(x for j in range(1, p2 + 1)
                           for x in ([p0 - j, p1], [p0 + j, p1], [p0, p1 - j], [p0, p1 + j]))]
    plus_cords.append(cords)

一旦我们有了它，我们也可以使外循环成为列表推导式。

plus_cords = [[[p0, p1], *(x for j in range(1, p2 + 1)
                             for x in ([p0 - j, p1], [p0 + j, p1], [p0, p1 - j], [p0, p1 + j]))]
              for p0, p1, p2 in pluses]

或者，您可以选择set 或tuples（以防止重复）而不是第二步，并以j=0 开头获取[p0, p1] 案例。

plus_cords = [{x for j in range(p2 + 1)
                 for x in ((p0 - j, p1), (p0 + j, p1), (p0, p1 - j), (p0, p1 + j))}
              for p0, p1, p2 in pluses]

Answer 2

一般来说，最好是迭代列表的元素而不是按索引迭代它，并且通过理解而不是迭代地构建一个新列表appending。

plus_cords = [[
    [p0, p1], [p0 - j, p1], [p0 + j, p1], [p0, p1 - j], [p0, p1 + j]
] for [p0, p1, p2] in pluses for j in range(1, p2 + 1)]

不是肯定的，因为我没有可供pluses 测试的样本数据，但这应该在一般情况下。