Python 高级特性（3）- 列表生成式

2021-04-19 17:56:14 浏览数 (1)

range() 函数

日常工作中，range() 应该非常熟悉了，它可以生成一个迭代对象，然后可以使用 list() 将它转成一个 list

代码语言：javascript复制
# 判断是不是迭代对象
print(isinstance(range(0, 10), Iterable))

# 生成列表
lists = list(range(0, 10))
print(lists)

# 输出结果
True
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
range() 的语法格式
代码语言：javascript复制
range(stop)
range(start, stop[, step])
range() 的作用

返回一个迭代对象，它是一个整型序列，从 start 开始，然后到 stop（不包括）结束，当然也可以设置它的步进 step

具体更多的栗子
代码语言：javascript复制
for i in range(5):
    print(i)

for i in range(10, 15):
    print(i)

for i in range(5, 25, 5):
    print(i)
运行结果
代码语言：javascript复制
0
1
2
3
4

10
11
12
13
14

5
10
15
20

列表生成式前置知识

  • 但如果要生成 怎么做？

[1x1, 2x2, 3x3, ..., 10x10]

  • 当你不会列表生成式前就是用循环
代码语言：javascript复制
res = []
for i in range(11):
    res.append(i * i)
print(res)

# 输出结果
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

但很明显繁琐了很多

列表生成式

针对上面的问题，列表生成式一句代码即可完成

代码语言：javascript复制
res = [i * i for i in range(11)]
print(res)

# 输出结果
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

列表生成式：带 if

代码语言：javascript复制
# 筛选偶数
res = [i * i for i in range(11) if i % 2 == 0]
print(res)

# 输出结果
[0, 4, 16, 36, 64, 100]
等价写法
代码语言：javascript复制
res = []
for i in range(11):
    if i % 2 == 0:
        res.append(i * i)

列表生成式：双重循环

代码语言：javascript复制
res = [i   j for i in range(5) for j in range(6, 11)]
print(res)

# 输出结果
[6, 7, 8, 9, 10, 7, 8, 9, 10, 11, 8, 9, 10, 11, 12, 9, 10, 11, 12, 13, 10, 11, 12, 13, 14]
等价写法
代码语言：javascript复制
res = []
for i in range(5):
    for j in range(6, 11):
        res.append(i   j)

列表生成式：多个变量

代码语言：javascript复制
d = {'x': 'A', 'y': 'B', 'z': 'C'}
res = [(k, v) for k, v in d.items()]
print(res)

# 输出结果
[('x', 'A'), ('y', 'B'), ('z', 'C')]
等价写法
代码语言：javascript复制
d = {'x': 'A', 'y': 'B', 'z': 'C'}
res = []
for k, v in d.items():
    res.append((k, v))

列表生成式：包含函数

代码语言：javascript复制
L = ['Hello', 'World', 'IBM', 'Apple']
res = [s.lower() for s in L]
print(res)

# 输出结果
['hello', 'world', 'ibm', 'apple']

列表生成式：包含 if...else

错误写法一
代码语言：javascript复制
res = [i * i for i in range(11) if i % 2 == 0 else 0]
print(res)

# 报错信息
res = [i * i for i in range(11) if i % 2 == 0 else 0]
　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　^
SyntaxError: invalid syntax
  • pycharm 也会直接提示语法错误
  • 因为跟在 for 后面的 if 是一个筛选条件，不能带 else
错误写法二
代码语言：javascript复制
res = [i if i % 2 == 0  for i in range(11)]
print(res)

# 报错信息
res = [i if i % 2 == 0  for i in range(11)]
                                  ^
SyntaxError: invalid syntax

for 前面是一个表达式，它需要有个表达式计算结果，只有 if 是不完整的，需要有 else

正确写法
代码语言：javascript复制
res = [i if i % 2 == 0 else "奇数" for i in range(11)]
print(res)

# 输出结果
[0, '奇数', 2, '奇数', 4, '奇数', 6, '奇数', 8, '奇数', 10]

将 if ... else 放在前面写

list pycharm range

0 人点赞