Python基础列表3——列表推导式

列表推导式的引入

我们如何把列表的每个元素都变成原来的两倍呢？
我们可以有两种办法

方法一：循环法

s = [1,2,3,4,5]
for i in s:
	s[i] = s[i]*2

这样的循环法会比较容易理解，但是由于涉及到了循环，我们代码的长度就会一定程度上的增加。不符合Python所倡导的简洁

方法二： 列表推导式

s = [1,2,3,4,5]
s = [i*2 for i in s]
s
>>>[2,4,6,8,10]

并且在效率方面，列表推导式的效率比循环语句的效率要高一倍

列表推导式的基本语法

[expression for target in iterable]
解释：expression是最后通过列表推导式所存放在列表中的元素
for…in…是for循环语句
target是和expression所对应的循环变量
iterable是一个可迭代的对象，在列表推导式中经常是列表

举例1

x = [i for i in range(10)]
x
>>>[0,1,2,3,4,5,6,7,8,9]

ps：在range中，range(A)的意思是输出0到A-1的数字注意不包括A

举例2

x = [i+1 for i in range(10)]
x
>>>[1,2,3,4,5,6,7,8,9,10]

举例3

x = [i*2 for i in "EchoMe"]
x
>>>['EE','cc','hh','oo','MM','ee']

字符串也是一个可迭代的对象

举例4

我们可以用列表推导式输出一个矩阵的主对角线的元素

matrix = [[1,2,3],[4,5,6],[7,8,9]]
diag = [matrix[i][i] for i in range(len(matrix))]
diag
>>>[1,5,9]
       

列表推导式内置if

语法的基本形式

[expression for target in iterable if condition]

举例

s = [i for i in range(10) if i%2 == 0]
s
>>>[0,2,4,6,8]

执行顺序：先执行for语句，再执行if语句，最后再执行最左边的表达式

可以利用列表推导式对列表中的元素的首字母进行分类

这里利用小甲鱼的举例：

words =["Great","FishC","Brillant","Excellent","Fantistic"]
fwords = [w for w in words if w[0] == 'F']
fwords
>>>[FishC,Fantastic]