Python列表操作-推导式（List Comprehension)

1.1 列表推导式的一般情况

列表推导式的一般语法结构：

new_list = [x for x in iterable]

**其中的iterable表示可迭代的对象，包括字符串（str）、列表（list）、元组（tuple）、字典（dict）、集合（set），以及生成器（generator）等。

下面举例说明：

1.1.1 迭代对象为字符串str

# 迭代对象为字符串strstr_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 迭代对象为列表

# 迭代对象为列表listlist_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