python - 将嵌套的 for 循环转换为列表 comp 并从结果列表中过滤完全平方数

Question

我编写了以下代码:

a_list = []

for x in range(5):
    a_list.append(dict())
    for y in range(5):
        if (x != 0 and y != 0) and (x * x != x * y):
            a_list[-1][y] = x * y

结果是:

[{}, {2: 2, 3: 3, 4: 4}, {1: 2, 3: 6, 4: 8}, {1: 3, 2: 6, 4: 12}, {1: 4, 2: 8, 3: 12}]

但是，我必须使用列表理解来获得相同的结果。请问我该怎么做？

Answer 1

以下嵌套理解将起作用:

[{y: x*y for y in range(5) if y and x*x != x*y} for x in range(5)]
# [{}, {2: 2, 3: 3, 4: 4}, {1: 2, 3: 6, 4: 8}, {1: 3, 2: 6, 4: 12}, {1: 4, 2: 8, 3: 12}]

内部字典理解的条件有点简化，因为 x*x != x*y 已经暗示了 x != 0。