Question

我需要使用列表理解来构建列表。这基本上就是它必须做的事情：

pattern = []
for c in range(3):
    for r in range(3):
        if r == c:
            pattern.append(a)
        else:
            pattern.append(b)

但是这一切都需要浓缩到一行！我以前从未使用过列表理解，所以请解释一下你的解决方案。

谢谢！

编辑：如果我希望新列表包含子列表，是否也可以将其列入列表理解中？上面我使用了3的范围，因此在生成的列表中，每个子列表将包含3个元素，即

pattern = [['a','b','b'],['b','a','b'],['b','b','a']]

Answer 1

列表理解的一般形式可以像这样理解

[(the task to be done) (how long it has to be done)]

我们通常在中使用for循环部分和要完成的任务部分可以if条件也是如此。 重要的是需要注意的是，要完成的任务部分应返回有效值（即使None是有效值）。因此，您不能在列表推导中使用任何Python语句（return，在Python 2.x print等中）。

现在回答你的第一个问题，

['a' if r == c else 'b' for c in range(3) for r in range(3)]
# ['a', 'b', 'b', 'b', 'a', 'b', 'b', 'b', 'a']

这正好创建了一个在for循环版本中显示的列表。

'a' if r == c else 'b'

这与

相同

if r == c:
    'a'
else:
    'b'

首先，for c in range(3)将被执行，并且将生成列表[0, 1, 2]，然后在每次迭代for r in range(3)将被执行并且将生成列表[0, 1, 2]。在r的每次迭代中，将执行上面看到的if条件，并且if..else的结果将用作生成的新列表的元素。

要回答第二个问题，您可以很好地使用列表理解。

我们从上面的例子中得到的基本理解是，列表理解是生成一个列表。现在，让我们尝试嵌套列表推导（我们将在要完成的任务部分中使用列表理解），就像这样

[['a' if r == c else 'b' for r in range(3)] for c in range(3)]
# [['a', 'b', 'b'], ['b', 'a', 'b'], ['b', 'b', 'a']]

现在，如果你看看我们的嵌套列表理解，首先，for c in range(3)将被执行，然后['a' if r == c else 'b' for r in range(3)]部分将被执行，这将生成嵌套列表的各个行。这里需要注意的重要一点是，c在嵌套列表推导中可用。