举例说明列表推导式如何使用?
for表达式:
1. a_range = range(10)
2. # 对a_range执行for表达式
3. a_list = [x * x for x in a_range]
4. # a_list集合包含10个元素
5. print(a_list)
[0 , 1 , 4 , 9 , 16 , 25 , 36 , 49 , 64, 81]
第 3 行会对 a_range 执行迭代,由于 a_range 相当于包含 10 个元素,因此程序生成的 a_list 同样包含 10 个元素,且每个元素都是 a_range 中每个元素的平方(由表达式 x*x 控制)。
还可以在列表推导式中添加 if 条件语句,这样列表推导式将只迭代那些符合条件的元素。
1. b_list = [x * x for x in a_range if x % 2 == 0]
2. # a_list集合包含5个元素
3. print(b_list)
[0 ,4 , 16, 36, 64]
可使用多个循环,就像嵌套循环一样。
1. d_list = [(x, y) for x in range(5) for y in range(4)]
2. # d_list列表包含20个元素
3. print(d_list)
相当于
1. dd_list = []
2. for x in range(5):
3. for y in range(4):
4. dd_list.append((x, y))
[(0, 0), (0, 1), (0, 2), (0, 3), (1, 0), (1, 1), (1, 2), (1, 3), (2, 0), (2, 1), (2, 2), (2, 3), (3, 0), (3, 1), (3, 2), (3, 3), (4, 0), (4, 1), (4, 2), (4, 3)]
1. src_a = [30, 12, 66, 34, 39, 78, 36, 57, 121]
2. src_b = [3, 5, 7, 11]
3. # 只要y能整除x,就将它们配对在一起
4. result = [(x, y) for x in src_b for y in src_a if y % x == 0]
5. print(result)
[(3, 30), (3, 12), (3, 66), (3, 39), (3, 78), (3, 36), (3, 57), (5, 30), (11, 66), (11, 121)]