11_Python的列表推导式_Python编程之路
上一节我们结束了有关python的方法相关内容
这一节我们先来学习python的列表推导式
之前跟大家说过range方法的使用,在这里我就不做过多的解释了,如忘记了,请翻阅之前的介绍
列表推导式,是Python内置的一种极其强大的生成list的表达式
如果要生成一个list [1 , 2 , 3 , 4 , 5 , 6 , 7 , 8 ] 可以用 range(1 , 9):
In [1]: list(range(1, 9)) Out[1]: [1, 2, 3, 4, 5, 6, 7, 8]
非常简单操作,相信大家一定都会
但我们要讲的并不是这个
所谓的列表推导式,就是指的轻量级的循环创建列表:
比如像下面的两个式子
In [2]: a = [i for i in range(1,10)] In [3]: a Out[3]: [1, 2, 3, 4, 5, 6, 7, 8, 9] In [4]: b = [10 for i in range(1, 10)] In [5]: b Out[5]: [10, 10, 10, 10, 10, 10, 10, 10, 10]
批量生成了两个列表,第一个是一个有顺序的从1到9的列表,第二个是生成了9个相同的数字10
简单讲一下这个列表推导式是怎么定义的
首先忽略推导式中的第一个字符,也就是 i 和 10 ,看后面的一段语句,很显然,是一个循环语句,之前讲过循环语句的,这里不详细解释了
然后再来观察最前面的字符,也就是列表中的元素,循环过的每一个都被作为列表的元素,如果你前面的字符是一个常数的话,那么就只需要考虑循环的次数,而不需要考虑循环的值
接下来,我们继续来加强这个列表推导式的应用
通过判断语句筛选列表元素
比如下面这段代码
In [6]: list_if = [i for i in range(1, 10) if i%2 == 0] In [7]: list_if Out[7]: [2, 4, 6, 8]
我们在里面加了判断条件,以获取给定循环范围内的所有的偶数
也就是说你可以在循环语句后面在加判断语句
这个还不能算是入门,接下来我们继续来加强列表推导式的应用
多个for循环
2个for循环
使用两个for循环语句,以获取二维列表
代码如下所示
In [8]: muti_for_list = [(i ,j) for i in range(1, 5) for j in range (6, 10)] In [9]: muti_for_list Out[9]: [(1, 6), (1, 7), (1, 8), (1, 9), (2, 6), (2, 7), (2, 8), (2, 9), (3, 6), (3, 7), (3, 8), (3, 9), (4, 6), (4, 7), (4, 8), (4, 9)]
这个列表推导式在一些算法领域会有涉及,我们通过连续写两个for循环语句,然后在开始以一个(i, j)作为获取的对象,在这里要注意以下,他是按照顺序执行的
三个for循环
获取三维数组
In [10]: muti_for_list = [(x, y, z) for x in range(2) for y in range (2) for z in range(2)] In [11]: muti_for_list Out[11]: [(0, 0, 0), (0, 0, 1), (0, 1, 0), (0, 1, 1), (1, 0, 0), (1, 0, 1), (1, 1, 0), (1, 1, 1)]
原理基本和二维数组的生成差不多,所以就不做过多解释了
在多维数组中使用判断语句
虽然我们在前面已经讲了多维数组的创建,但为了加强这个的复杂度,我们继续加一些条件
获取三维数组中每行组中元素都一样的数组 类似于下面这样的
[(1, 1, 1), (2, 2, 2), (3, 3, 3)]
下面我们通过代码来实现这个功能
In [12]: common_list = [(x, y, z) for x in range(10) for y in range (10) for z in range(10) if x == y and y == z] In [13]: common_list Out[13]: [(0, 0, 0), (1, 1, 1), (2, 2, 2), (3, 3, 3), (4, 4, 4), (5, 5, 5), (6, 6, 6), (7, 7, 7), (8, 8, 8), (9, 9, 9)]
这个并不难实现,无非就是加个判断条件
列表推导式,就讲到这里,他还有很多种其他的用法,希望大家,自己去挖掘,去探索
特克斯博客官网 www.susmote.com