形象地解释 Python 中的列表解析

529 查看

有时候,如果一种编程设计模式想要“捍卫”自己独特的语法形态不被改变,就会把自己变成普遍的流行趋势。Python 的列表推导就是这样一个典型的语法糖的例子。

Python 中的列表推导是非常棒的,但是要精通它们就比较困难了,因为它不是在解决一个新的问题:它只是提供了一种新的语法来解决一个已经存在的问题。

让我们来学习一下什么是列表推导,以及如何辨别什么时候该使用列表推导。

什么是列表推导?

列表推导是一个将一个列表(实际上是任意可迭代对象)转换成另一个列表的工具。在转换时,每个元素都可以按照某个条件被包含在新的列表中,并根据需要做出一些变换。

如果你熟悉函数式编程,你可以把列表推导想成是一个filter后面跟了一个map的语法糖

如果你不熟悉函数式编程,别担心,我会用for循环来解释。

从循环到列表推导

每个列表推导都可以重写成for循环的形式,但并不是每一个for循环都可以重写成列表推导。

要理解什么时候该使用列表推导,关键在于不断练习辨别哪些问题看起来像是列表推导。

如果你能把你的代码重写成这个样子的for循环,那你就能把它重写成列表推导:

你可以把上面的for循环重写成这个样子的列表推导:

列表推导:动态图™

看起来不错,不过具体要怎么做呢?

我们用复制粘贴来把一个for循环变成列表推导。

下面是复制粘贴的顺序:

  1. 复制给新列表赋值的语句(第三行)
  2. 把我们append到新列表的表达式复制过来(第六行)
  3. 复制for循环的那一行,除去末尾的:(第四行)
  4. 复制if语句,除去末尾的:(第五行)

现在我们就把下面这个for 循环:

变成了这个样子:

列表推导:现在加上颜色

我们来给代码加上高亮。

doubled_odds = []
    for n in numbers:
        if n % 2 == 1:
            doubled_odds.append(n * 2)
doubled_odds = [n * 2 for n in numbers if n % 2 == 1]

  1. 复制给新列表赋值的语句
  2. 把我们append到新列表的表达式复制过来
  3. 复制for循环的那一行,除去末尾的:
  4. 复制if语句,同样去掉:

无条件列表推导

但如果是没有条件语句(就是末尾的那个if SOMETHING)的情形呢?这些循环添加元素的for循环比我们刚刚讲过的那种循环并根据条件添加元素的要更简单。

没有if语句的for循环:

同样的代码写成列表推导: