各种推导式

发表于 | 分类于


推导式的套路:

  1. 新的列表 = [ 每一个元素或者是和元素相关的操作 for 元素 in 可迭代数据类型 ] # 遍历之后挨个处理
  2. 新的列表 = [ 满足条件的元素相关的操作 for 元素 in 可迭代数据类型 if 元素相关的条件 ] # 筛选功能

列表推导式

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
# 例一：30以内所有能被3整除的数
l = [ i for i in range(30) if i % 3 == 0 ]
print(l)

# 例二：30以内所有能被3整除的数的平方
l = [ i**2 for i in range(30) if i % 3 == 0 ]
print(l)

# 例三:找到嵌套列表中名字含有两个‘e’的所有名字
names = [['Tom', 'Billy', 'Jefferson', 'Andrew', 'Wesley', 'Steven', 'Joe'],
         ['Alice', 'Jill', 'Ana', 'Wendy', 'Jennifer', 'Sherry', 'Eva']]

# 常规循环
#for l in names:
#    for name in l:
#        if name.count('e') >= 2:
#            print(name)

# 二维列表
# 注意遍历顺序，这是实现的关键
l = [name for l in names for name in l if name.count('e') >= 2]
print(l)

字典推导式

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
# 字典推导式
# 例一：将一个字典的key和value对调
mcase = {'a': 10, 'b': 34}
# 常规循环
# for k in mcase:
#    mcase[k] : k    # v : k
# print(mcase)
mcase = {mcase[k]:k for k in mcase}
print(mcase)

# 例二：合并大小写对应的value值，将k统一成小写
# 所有的推导式都从for开始看
mcase = {'a': 10, 'b': 34, 'A': 7, 'Z': 3}
# {'a':10+7,'b':34,'z':3}
mcase_frequency = {k.lower(): mcase.get(k.lower(), 0) + mcase.get(k.upper(), 0) for k in mcase.keys()}
# 拿到所有key
# key值就是要小写k.lower():
# mcase.get(k.lower(), 0)没有小写的就默认写一个0 + mcase.get(k.upper(), 0) 相当于小写+大写的值,没有就给0
print(mcase_frequency)

集合推导式

1
2
3
4
5
# 集合推导式 可以替重
# {}
# 例：计算列表中每个值的平方，自带去重功能
squared = {x**2 for x in [1, -1, 2]}
print(squared)  # -1的平方和1的平方一样 所以去重之后2个结果，如果是列表就不会自动去重

总结

各种推导式:生成器 列表 字典 集合

  1. 遍历
  2. 筛选 带if

优点

  1. 推导式能让你的代码更简洁，
  2. 可读性提高

惰性运算:

  1. 懒
  2. 不去找他要值不会执行，要一个执行一个
  3. 生成器(自己写的，可见)和迭代器（不可见的，python提供）
    特点
  4. 同一个迭代器,从头到尾取值只能取一次
  5. 不找他要值的时候不干活

练习

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
# 例1:  过滤掉长度小于3的字符串列表，并将剩下的转换成大写字母
names = ['Tom', 'Billy', 'Jefferson', 'Andrew', 'Wesley', 'Steven', 'Joe']
new_l = [name.upper() for name in names if len(name) > 3]
print(new_l)

# 例2:  求(x,y)其中x是0-5之间的偶数，y是0-5之间的奇数组成的元祖列表
l1 = [(x,y) for x in range(5) if x%2==0 for y in range(5) if y %2==1]
l2 = [(x,y) for x in range(5) if x % 2 == 0 for y in range(5) if y % 2 == 1]
print(l1)
print(l2)

# 求M中3,6,9组成的列表
M = [[1,2,3],[4,5,6],[7,8,9]]
# for i in M:
#     print(i[2])
l3 = [row[2] for row in M]
print(l3)