“清单理解”是什么意思？它是如何工作的，我如何使用它？

Question

[x ** 2 for x in range(10)]

[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

我搜索过，似乎这被称为列表理解，但它是如何工作的？

Answer 1

从文档中：

列表理解提供了创建列表的简明方法。常见的应用程序是创建新的列表，其中每个元素是应用于另一个序列的每个成员的一些操作的结果或可迭代的，或者创建满足特定条件的这些元素的子序列。

关于您的问题，列表理解与下面的“普通”Python代码做了相同的事情：

>>> l = [] 
>>> for x in range(10):
...     l.append(x**2)
>>> l
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

>>> list(map(lambda x: x**2, range(10)))
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

>>> [x**2 for x in range(10)]
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

>>> [x.strip() for x in ('foo\n', 'bar\n', 'baz\n')]
['foo', 'bar', 'baz']

>>> [int(x) for x in ('1', '2', '3')]
[1, 2, 3]

>>> d = {'foo': '10', 'bar': '20', 'baz': '30'}
>>> [d[x] for x in ['foo', 'baz']]
['10', '30']

>>> d = {'foo': '10', 'bar': '20', 'baz': '30'}
>>> [int(d[x].rstrip('0')) for x in ['foo', 'baz']]
[1, 3]

>>> l = []
>>> for x in range(10):
...     if x%2:
...         l.append(x)
>>> l
[1, 3, 5, 7, 9]

啊，那太复杂了。下面的版本呢？

>>> [x for x in range(10) if x%2]
[1, 3, 5, 7, 9]

若要使用if...else三元表达式，需要将if...else...放在x之后，而不是放在范围(10)之后：

>>> [i if i%2 != 0 else None for i in range(10)]
[None, 1, None, 3, None, 5, None, 7, None, 9]

你听说过嵌套列表理解吗？您可以将两个或多个for放在一个列表理解中。例如：

>>> [i for x in [[1, 2, 3], [4, 5, 6]] for i in x]
[1, 2, 3, 4, 5, 6]

>>> [j for x in [[[1, 2], [3]], [[4, 5], [6]]] for i in x for j in i]
[1, 2, 3, 4, 5, 6]

让我们讨论第一部分，[[1,2,3],[4,5,6]]中x的，它给出[1,2,3]和[4,5,6]。然后，for i in x给出1、2、3和4、5、6。

警告：您总是需要将x in[[1,2,3],[4,5,6]]的放在i in x的之前：

>>> [j for j in x for x in [[1, 2, 3], [4, 5, 6]]]
Traceback (most recent call last):
  File "<input>", line 1, in <module>
NameError: name 'x' is not defined

>>> {x for x in [1, 1, 2, 3, 3, 1]}
{1, 2, 3}

>>> set([i for i in [1, 1, 2, 3, 3, 1]])
{1, 2, 3}

>>> {i: i**2 for i in range(5)}
{0: 0, 1: 1, 2: 4, 3: 9, 4: 16}

>>> d = {}
>>> for i in range(5):
...     d[i] = i**2
>>> d
{0: 0, 1: 1, 2: 4, 3: 9, 4: 16}

(i表示范围（5）中的i）是否给出元组？不！，这是一个生成器表达式。返回生成器：

>>> (i for i in range(5))
<generator object <genexpr> at 0x7f52703fbca8>

与：

>>> def gen():
...     for i in range(5):
...         yield i
>>> gen()
<generator object gen at 0x7f5270380db0>

您可以将其用作生成器：

>>> gen = (i for i in range(5))
>>> next(gen)
0
>>> next(gen)
1
>>> list(gen)
[2, 3, 4]
>>> next(gen)
Traceback (most recent call last):
  File "<input>", line 1, in <module>
StopIteration

>>> sum(i**2 for i in range(5))
30