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