python列表推导式

偶尔会在python中看见这样一行代码：

data = [x**2 for x in range(0, 5)]
# 此时data = [0, 1, 4, 9, 16]

如果不知道背后的语法规则，那么很难理解为什么可以在列表的方括号中写一个“for”循环？实际上，这种语法称为列表推导式，和lambda表达式类似，都是在代码很短的情况下被使用。

本文将从两个方面讲解列表推导式：

• （1）什么是列表推导式
• （2）如何使用列表推导式

1、什么是列表推导式？

列表推导式对应的英文是list comprehension，有时也被翻译为列表解析式，是一种创建列表的简洁语法。在开始分析它之前，

先看下面这种较为常用的列表创建方式：

data = []   # 创建空列表
for x in range(-5, 5):
    if x >= -2: 
        # 如果x>=-2则给data添加i的平方
        data.append(x**2) 
print(data)
#　输出：[4, 1, 0, 1, 4, 9, 16]

上面的代码首先创建了一个空列表，然后通过for循环来给列表添加新元素，并且需要通过if语句保证x在一定的数值范围，这是一种较为常见的创建方式。

问题在于，上述的创建需要写好几行代码，那么能不能提供一种简洁的创建列表的方式呢？列表推导式就是起到这个功能的，

它的原理图如下：

它的语法分成3个部分：

• （1）需要多少个元素：x的取值为从-5到5，也就是需要循环9次
• （2）判断本次循环是否插入新的元素：虽然一共需要9次循环，但并不是每次循环都必须要插入一个新元素（列表不一定要包含9个数值），每次循环时需要按照某种判断条件，比如当前循环的x是否大于-2，如果大于-2则插入一个新的值，如果小于-2则不插入新的值。
• （3）元素的值：当前循环想要插入某个值，这个值可以是包含x的某个表达式，也可以不包含x。

现在，我们可以使用列表推导式来实现刚才的功能了：

data = [x**2 for x in range(-5, 5) if x >= -2]
print(data)
# 输出：[4, 1, 0, 1, 4, 9, 16]

可以把上面的三个部分称为：循环次数、想插入的值、判断是否插入的条件

2、如何使用列表推导式

前面的介绍提到：列表推导式包含三个部分，分别是循环次数、想插入的值、判断是否插入的条件。现在，使用这三个部分来学习如何使用三个列表推导式，来用简洁的列表创建代码。

题目：用15以内的奇数创建列表

分析：

• （1）循环次数：15
• （2）想插入的值：奇数
• （3）判断条件：是不是奇数

所以写出了下面的代码：

得到了下面的值：

[1, 3, 5, 7, 9, 11, 13, 15]

总结：

列表推导式虽然看起来不符合常规，因为它在方括号内部使用for循环，但是它却起到了让代码更加简洁的作用，尤其是它可以通过“判断条件”来决定什么情况下插入新的值。