Python3 列表、字典、元组、集合解析(推导式)各种写法

https://blog.csdn.net/weixin_41611054/article/details/113870870

因为解析式的高效和简洁，所以这里记录下列表、字典、元组、集合解析的各种写法。

列表解析

通过[]或list()进行列表解析。

单层for循环

1. data = [1, 2, 3, 4, 5, 6, 7]
3. a = [i for i in data]
4. # 或者：a = list(i for i in data)
6. # 执行结果：[1, 2, 3, 4, 5, 6, 7]
7. # 执行顺序：
8. a = []
9. for i in data:
10.     a.append(i)

单层for循环 + if判断(if写在后面)

1. data = [1, 2, 3, 4, 5, 6, 7]
3. b = [i for i in data if i != 1]
4. # 或者：b = list(i for i in data if i != 1)
6. # 执行结果：[2, 3, 4, 5, 6, 7]
7. # 执行顺序：
8. b = []
9. for i in data:
10.     if i != 1:
11.         b.append(i)

单层for循环 + if判断 + else(if写在前面)

1. data = [1, 2, 3, 4, 5, 6, 7]
3. c = [i if i != 1 else 0 for i in data]
4. # 或者：c = list(i if i != 1 else 0 for i in data)
6. # 执行结果：[0, 2, 3, 4, 5, 6, 7]
7. # 执行顺序：
8. c = []
9. for i in data:
10.     if i != 1:
11.         c.append(i)
12.     else:
13.         c.append(0)

多层for循环

1. data = [1, 2, 3, 4, 5, 6, 7]
3. d = [i * j for i in data for j in range(3)]
4. # 或者：d = list(i * j for i in data for j in range(3))
6. # 执行结果：[0, 1, 2, 0, 2, 4, 0, 3, 6, 0, 4, 8, 0, 5, 10, 0, 6, 12, 0, 7, 14]
7. # 执行顺序：
8. d = []
9. for i in data:
10.     for j in range(3):
11.         d.append(i * j)

多层for循环 + if判断

1. data = [1, 2, 3, 4, 5, 6, 7]
3. e = [i * j for i in data for j in range(3) if i == j]
4. # 或者：e = list(i * j for i in data for j in range(3) if i == j)
6. # 执行结果：[1, 4]
7. # 执行顺序：
8. e = []
9. for i in data:
10.     for j in range(3):
11.         if i == j:
12.             e.append(i * j)

多层for循环 + 多个if判断(多个if可用and连接)

1. data = [1, 2, 3, 4, 5, 6, 7]
3. f = [i * j for i in data for j in range(3) if i > 1 if j > 1]
4. # 或者：f = list(i * j for i in data for j in range(3) if i > 1 if j > 1)
6. # 执行结果：[4, 6, 8, 10, 12, 14]
7. # 执行顺序：
8. f = []
9. for i in data:
10.     for j in range(3):
11.         if i > 1:
12.             if j > 1:
13.                 f.append(i * j)
15. f = [i * j for i in data for j in range(3) if i > 1 and j > 1]
16. # 或者：f = list(i * j for i in data for j in range(3) if i > 1 and j > 1)
18. # 执行结果：[4, 6, 8, 10, 12, 14]
19. # 执行顺序：
20. f = []
21. for i in data:
22.     for j in range(3):
23.         if i > 1 and j > 1:
24.                 f.append(i * j)

字典解析

通过{}或dict()进行字典解析。

单层for循环

1. data = {'a': 0, 'b': 1, 'c': 2, 'd': 3}
3. a = {v: k for k, v in data.items()}
4. # 或者：a = dict((v, k) for k, v in data.items())
6. # 执行结果：{0: 'a', 1: 'b', 2: 'c', 3: 'd'}
7. # 执行顺序：
8. a = {}
9. for k, v in data.items():
10.     a.update({v: k})

单层for循环 + if判断(if写在后面)

1. data = {'a': 0, 'b': 1, 'c': 2, 'd': 3}
3. b = {k: v for k, v in data.items() if v > 1}
4. # 或者：b = dict((k, v) for k, v in data.items() if v > 1)
6. # 执行结果：{'c': 2, 'd': 3}
7. # 执行顺序：
8. b = {}
9. for k, v in data.items():
10.     if v > 1:
11.         b.update({k: v})

单层for循环 + if判断 + else(if写在前面)

1. data = {'a': 0, 'b': 1, 'c': 2, 'd': 3}
3. c = {k: v if v > 1 else None for k, v in data.items()}
5. # 执行结果：{'a': None, 'b': None, 'c': 2, 'd': 3}
6. # 执行顺序：
7. c = {}
8. for k, v in data.items():
9.     if v > 1:
10.         c.update({k: v})
11.     else:
12.         c.update({k: None})

多层for循环

1. data = {'a': {'num0': 0}, 'b': {'num1': 1}, 'c': {'num2': 2}, 'd': {'num3': 3}}
3. d = {k1: v2 for k1, v1 in data.items() for k2, v2 in v1.items()}
4. # 或者：d = dict((k1, v2) for k1, v1 in data.items() for k2, v2 in v1.items())
6. # 执行结果：{'a': 0, 'b': 1, 'c': 2, 'd': 3}
7. # 执行顺序：
8. d = {}
9. for k1, v1 in data.items():
10.     for k2, v2 in v1.items():
11.         d.update({k1: v2})

多层for循环 + if判断

1. data = {'a': {'num0': 0}, 'b': {'num1': 1}, 'c': {'num2': 2}, 'd': {'num3': 3}}
3. e = {k1: v2 for k1, v1 in data.items() for k2, v2 in v1.items() if k2 != 'num0'}
4. # 或者：d = dict((k1, v2) for k1, v1 in data.items() for k2, v2 in v1.items() if k2 != 'num0')
6. # 执行结果：{'b': 1, 'c': 2, 'd': 3}
7. # 执行顺序：
8. e = {}
9. for k1, v1 in data.items():
10.     for k2, v2 in v1.items():
11.         if k2 != 'num0':
12.             e.update({k1: v2})

多层for循环 + 多个if判断(多个if可用and连接)

1. data = {'a': {'num0': 0}, 'b': {'num1': 1}, 'c': {'num2': 2}, 'd': {'num3': 3}}
3. f = {k1: v2 for k1, v1 in data.items() for k2, v2 in v1.items() if k2 != 'num0' if v2 > 1}
4. # 或者：f = dict((k1, v2) for k1, v1 in data.items() for k2, v2 in v1.items() if k2 != 'num0' if v2 > 1)
6. # 执行结果：{'c': 2, 'd': 3}
7. # 执行顺序：
8. f = {}
9. for k1, v1 in data.items():
10.     for k2, v2 in v1.items():
11.         if k2 != 'num0':
12.             if v2 > 1:
13.                 f.update({k1: v2})
15. f = {k1: v2 for k1, v1 in data.items() for k2, v2 in v1.items() if k2 != 'num0' and v2 > 1}
16. # 或者：f = dict((k1, v2) for k1, v1 in data.items() for k2, v2 in v1.items() if k2 != 'num0' and v2 > 1)
18. # 执行结果：{'c': 2, 'd': 3}
19. # 执行顺序：
20. f = {}
21. for k1, v1 in data.items():
22.     for k2, v2 in v1.items():
23.         if k2 != 'num0' and v2 > 1:
24.                 f.update({k1: v2})

元组解析

通过()或tuple()进行元组解析，前者返回的是元组，后者返回的是个生成器。生成器占用的内存较小，可通过循环读取。写法和列表解析相似。

1. data = [1, 2, 3, 4, 5, 6, 7]
3. ## 通过()解析
4. a = (i for i in data)
5. # 执行结果：<generator object <genexpr> at 0x7f1e24bf1fc0>
7. ## 单层for循环
8. a = tuple(i for i in data)
9. # 执行结果：(1, 2, 3, 4, 5, 6, 7)
11. ## 单层for循环 + if判断(if写在后面)
12. b = tuple(i for i in data if i != 1)
13. # 执行结果：(2, 3, 4, 5, 6, 7)
15. ## 单层for循环 + if判断 + else(if写在前面)
16. c = tuple(i if i != 1 else 0 for i in data)
17. # 执行结果：(0, 2, 3, 4, 5, 6, 7)
19. ## 多层for循环
20. d = tuple(i * j for i in data for j in range(3))
21. # 执行结果：(0, 1, 2, 0, 2, 4, 0, 3, 6, 0, 4, 8, 0, 5, 10, 0, 6, 12, 0, 7, 14)
23. ## 多层for循环 + if判断
24. e = tuple(i * j for i in data for j in range(3) if i == j)
25. # 执行结果：(1, 4)
27. ## 多层for循环 + 多个if判断(多个if可用and连接)
28. f = tuple(i * j for i in data for j in range(3) if i > 1 if j > 1)
29. f = tuple(i * j for i in data for j in range(3) if i > 1 and j > 1)
30. # 执行结果：(4, 6, 8, 10, 12, 14)

集合解析

通过{}或set()进行集合解析。集合中没有重复元素。写法和列表解析相似。

1. data = [1, 2, 3, 4, 5, 6, 7, 7, 1]
3. ## 通过{}解析
4. a = {i for i in data}
5. # 执行结果：{1, 2, 3, 4, 5, 6, 7}
7. ## 单层for循环
8. a = set(i for i in data)
9. # 执行结果：{1, 2, 3, 4, 5, 6, 7}
11. ## 单层for循环 + if判断(if写在后面)
12. b = set(i for i in data if i != 1)
13. # 执行结果：{2, 3, 4, 5, 6, 7}
15. ## 单层for循环 + if判断 + else(if写在前面)
16. c = set(i if i != 1 else 0 for i in data)
17. # 执行结果：{0, 2, 3, 4, 5, 6, 7}
19. ## 多层for循环
20. d = set(i * j for i in data for j in range(3))
21. # 执行结果：{0, 1, 2, 3, 4, 5, 6, 7, 8, 10, 12, 14}
23. ## 多层for循环 + if判断
24. e = set(i * j for i in data for j in range(3) if i == j)
25. # 执行结果：{1, 4}
27. ## 多层for循环 + 多个if判断(多个if可用and连接)
28. f = set(i * j for i in data for j in range(3) if i > 1 if j > 1)
29. f = set(i * j for i in data for j in range(3) if i > 1 and j > 1)
30. # 执行结果：{4, 6, 8, 10, 12, 14}