Python 什么是；列表理解“；什么意思？它是如何工作的，我如何使用它？

我有以下代码：

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

当我在Python shell中运行它时，它返回：

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

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

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

---

关于您的问题，列表理解与以下“普通”Python代码的作用相同：

>>> l = [] 
>>> for x in range(10):
...     l.append(x**2)
>>> l
[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

做任何事情。不仅仅是

x**2

。例如，运行

x

的方法：

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

>>> [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

作为另一个函数的参数：

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

例如，我们还可以使用

x

作为

dict

对象的键。让我们看看：

>>> 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]

等等

---

您还可以在列表中使用

if

或

if…else

。例如，您只需要

范围（10）

中的奇数。你可以做：

>>> 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。。。否则…

在

x

之后，不在

范围（10）

之后：

---

你听说了吗？您可以在一个列表中为s放置两个或多个

。例如：

>>> [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]

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

让我们来谈谈第一部分，[[1,2,3]，[4,5,6]

中x的

，它给出了
[1,2,3]
和
[4,5,6]
。然后，x中i的
给出
1
，
2
，
3
和
4
，
5
，
6

警告：在[1,2,3]、[4,5,6]、[code>之前，始终需要将x的
放在[[1,2,3]、[4,5,6]中

>>> [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

这与：

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

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

A看起来像一个集合理解，但它使用
{key:value for key，value in…}
或
{i:i for i in…}
而不是
{i for i in…}

例如：

>>> [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]

>>> {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}

（范围（5）中的i表示i）
是否给出一个元组？不这是一个好主意。它返回一个生成器：

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

相关的（关于生成器）：。
有列表、字典和集合理解，但没有元组理解（尽管确实探索了“生成器表达式”）

它们解决了Python中传统循环是语句（不返回任何内容）而不是返回值的表达式的问题

它们不是所有问题的解决方案，可以重写为传统循环。当需要在迭代之间维护和更新状态时，它们会变得笨拙

它们通常包括：

[<output expr> <loop expr <input expr>> <optional predicate expr>]

[]

但是可以用很多有趣和奇怪的方式扭曲

它们可以类似于传统的
map（）
和
filter（）
操作，这些操作仍然存在于Python中并继续使用

如果做得好，他们会有很高的满意度。
如果你更喜欢用一种更直观的方式来了解发生了什么，那么这可能会有帮助：

# for the example in the question...

y = []
for x in range(10):
    y += [x**2]

# is equivalent to...

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

# for a slightly more complex example, it is useful
# to visualize  where the various x's end up...

a = [1,2,3,4]
b = [3,4,5,6]
c = []

for x in a:
          if x in b:
                  c += [x]
#   \         \        /
#    \    _____\______/
#     \  /      \
#      \/        \
#      /\         \
#     /  \         \
#    /    \         \
c = [x for x in a if x in b]

print(c)

…生成输出
[3,4]
最近，我看到了很多关于列表理解如何工作的困惑（在其他SO问题上以及同事的问题上）。一点点的数学教育可以帮助理解为什么语法是这样的，以及列表理解的真正含义

语法
最好将列表理解看作集合/集合上的谓词，就像我们在数学中使用集合生成器表示法一样。这个符号对我来说真的很自然，因为我拥有数学学士学位。但是忘了我吧，Guido van Rossum（Python的发明者）拥有数学硕士学位，有数学背景

速成班
下面是集合生成器表示法的工作原理（非常基础）：


因此，这个集合生成器符号表示严格正的一组数字（即
[1,2,3,4，…]
）

混淆点
1） 集合生成器表示法中的谓词过滤器仅指定要保留的项，列表理解谓词也执行相同的操作。您不必包含用于省略项的特殊逻辑，除非谓词包含这些项，否则它们将被省略。空谓词（即末尾没有条件）包括给定集合中的所有项

2） 集合生成器表示法中的谓词过滤器位于末尾，在列表理解中也是如此。（一些）初学者认为类似于
[x<5 for x in range（10）]
的东西会给他们列表
[0,1,2,3,4]
，而实际上它会输出
[True，True，True，True，True，False，False，False]
。我们得到了输出
[True，True，True，True，False，False，False，False，False]
，因为我们要求Python为
范围（10）
中的所有项计算
x<5
。无谓词意味着我们从集合中获取所有内容（就像集合生成器符号一样）

如果在使用列表理解时将集合生成器符号放在脑后，它们会更容易接受。

嗯
 简介

列表理解是在Python中创建列表的一种高级声明方式。理解的主要好处是可读性和可维护性。很多人觉得它们可读性很强，甚至以前从未见过它们的开发人员通常也能正确地猜出它的意思

#片段1
平方=[n**2表示范围内的n（5）]
#片段2
正方形=[]
对于
[<output expr> <loop expr <input expr>> <optional predicate expr>]

# for the example in the question...

y = []
for x in range(10):
    y += [x**2]

# is equivalent to...

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

# for a slightly more complex example, it is useful
# to visualize  where the various x's end up...

a = [1,2,3,4]
b = [3,4,5,6]
c = []

for x in a:
          if x in b:
                  c += [x]
#   \         \        /
#    \    _____\______/
#     \  /      \
#      \/        \
#      /\         \
#     /  \         \
#    /    \         \
c = [x for x in a if x in b]

print(c)

>>> type(squares)
<class 'list'>