偶尔会在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循环,但是它却起到了让代码更加简洁的作用,尤其是它可以通过“判断条件”来决定什么情况下插入新的值。
到此这篇关于python列表推导式的原理及使用方法的文章就介绍到这了,更多相关python列表推导式内容请搜索编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程网!