Python列表操作-推导式(List Comprehension)
1.1 列表推导式的一般情况
列表推导式的一般语法结构:
new_list = [x for x in iterable]
**其中的iterable表示可迭代的对象,包括字符串(str)、列表(list)、元组(tuple)、字典(dict)、集合(set),以及生成器(generator)等。
下面举例说明:
1.1.1 迭代对象为字符串str
# 迭代对象为字符串str
str_list = [x.lower() for x in 'Bruce_liuxiaowei']
str_list1 = [x.upper() for x in 'burce_liuxiaowei']
print(str_list, str_list1, end='\n')
输出结果:
['b', 'r', 'u', 'c', 'e', '_', 'l', 'i', 'u', 'x', 'i', 'a', 'o', 'w', 'e', 'i']
['B', 'R', 'U', 'C', 'E', '_', 'L', 'I', 'U', 'X', 'I', 'A', 'O', 'W', 'E', 'I']
1.1.2 迭代对象为列表
# 迭代对象为列表list
list_list = [x ** 3 for x in [4, 5, 6, 7]]
print(list_list)
输出结果:
[64, 125, 216, 343]
1.1.3 迭代对象为元组
# 迭代对象为元组
tuple_list = [x+2 for x in (1, 2, 3, 4)]
print(tuple_list)
tuple_list = [x-4 for x in (1, 2, 3, 4)]
print(tuple_list)
输出结果:
[3, 4, 5, 6]
[-3, -2, -1, 0]
1.1.4 迭代对象为集合
# 迭代对象为集合
tuple_list = [x+3 for x in (1, 2, 3, 4)]
print(tuple_list)
tuple_list = [x-3 for x in (1, 2, 3, 4)]
print(tuple_list)
输出结果:
[4, 5, 6, 7]
[-2, -1, 0, 1]
1.1.5 迭代对象为生成器(generator)
# 迭代器为生成器(generator)
ge_list = [x for x in range(10)]
print(ge_list)
输出结果:
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
1.2 两层for循环的列表推导式
列表推导式中,可以同时包含多个for循环,比如同时两个for循环,举例说明:
1.2.1 代码:
print([x**2+y for x in range(8) for y in range(3, 6)])
输出结果:
[3, 4, 5, 4, 5, 6, 7, 8, 9, 12, 13, 14, 19, 20, 21, 28, 29, 30, 39, 40, 41, 52, 53, 54]
上述结果与下面的写法输出结果一样的
two_for_list = []
for x in range(8):
for y in range(3, 6):
two_for_list.append(x**2 + y)
print(two_for_list)
输出结果:
[3, 4, 5, 4, 5, 6, 7, 8, 9, 12, 13, 14, 19, 20, 21, 28, 29, 30, 39, 40, 41, 52, 53, 54]
备注:列表推导式中,三层或者三层以上的for循环很少用
1.3 使用两个变量来生成list
列表推导式也可以使用两个或多个变量来生成list,结合字典的使用,举例如下:
1.3.1 示例代码:
d = {
'x': '4', 'y': '5', 'z': '6'}
d_list = [k+'='+v for k, v in d.items()]
print(d_list)
输出结果:
['x=4', 'y=5', 'z=6']
1.4 含if语句的列表推导式
列表推导式中还可以引入if条件语句,举例如下:
1.4.1 示例代码:
if_list = [x**3 for x in range(8) if x % 2 == 0]
print(if_list)
输出结果:
[0, 8, 64, 216]
上述列表推导式,如果用普通的for循环来完成,编写如下:
# 普通for循环
if_list_1 = []
for i in range(8):
if i % 2 == 0:
if_list_1.append(i**3)
print(if_list_1)
输出结果:
[0, 8, 64, 216]
通过对比,可以看出列表推导式的方式更显得Pythonic。当然,如果仅仅是编写上更简洁,可能不一定能显现出列表推导式的优势,下面我们来对比一下上述两种方式的运行时间效率。
从上面的运行结果进行对比,可以看出列表推导式的运行效率要高于普通的for循环编写方式。所以建议使用列表推导式,不仅写法更Pythonic,且运行效率更高。
1.5 包含if-else语句的列表推导式
1.5.1 示例代码:
if_else_list = [x ** 3 if x%2 == 0 else x+3 for x in range(10)]
print(if_else_list)
输出结果:
[0, 4, 8, 6, 64, 8, 216, 10, 512, 12]
1.6 包含两个if语句的列表推导式
1.6.1示例代码:
double_list = [x**2 for x in range(10) if x % 2==0 if x %3 ==0]
print(double_list)
输出结果:
[0, 36]
总结
如果想获取离散的元素,比如想获取第1、2、4个元素,能不能通过离散的索引值来获取呢?答案是否定的。
示例代码如下:
names = ['James', 'Michael', 'Emma', 'Emily']
print(names[1, 2, 4])
输出结果:
Traceback (most recent call last):
File "/Users/liuxiaowei/PycharmProjects/Python基础教程/Python知识手册/tutorials/Python数据类型-List介绍/列表推导式/小结.py", line 17, in <module>
print(names[1, 2, 4])
TypeError: list indices must be integers or slices, not tuple