python推导式

在python基础100道中，看到python推导式，不解这是个啥，了解语法后发现还挺有意思的。

例如：

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

 

 

 相当于：

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]

 

相关语法：

它的语法分成3个部分：

（1）需要多少个元素：x的取值为从-5到5，也就是需要循环9次

（2）判断本次循环是否插入新的元素：虽然一共需要9次循环，但并不是每次循环都必须要插入一个新元素（列表不一定要包含9个数值），每次循环时需要按照某种判断条件，比如当前循环的x是否大于-2，如果大于-2则插入一个新的值，如果小于-2则不插入新的值。

（3）元素的值：当前循环想要插入某个值，这个值可以是包含x的某个表达式，也可以不包含x。

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

 

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

1.循环次数：15次

2.想插入的值：奇数

3.判断是否插入的条件：判断是否为奇数

data_list = [x for x in range(16) if x%2==1]
print(data_list)
# data_list = [1, 3, 5, 7, 9, 11, 13, 15]

 

参考：https://zhuanlan.zhihu.com/p/90558097