Python 列表推导式、生成器表达式

列表推导式

一行代码几乎搞定你需要的任何的列表。

• 循环模式
  • [变量(或加工后的变量) for 变量 in iterable]
  • l = [i for i in range(1, 101)]
  • l = [i*i for i in range(1, 6)]
• 筛选模式
  • [变量(或加工后的变量) for 变量 in iterable if 条件]
  • l = [i for i in range(1, 31) if i % 2 == 0]
  • 两层循环
    • l1 = [['tom', 'jack', 'lily'], ['nancy', 'lucy', 'stank', 'jense']]
    • l = [j for i in l1 for j in i if j.count("n") == 2]

优缺点

• 优点：一行解决，方便。
• 缺点：容易着迷，不易排错。
• 列表推导式不能解决所有列表的问题，所以不要太刻意用。

生成器表达式

将列表推导式的 [] 换成 () 即可。

# 100W数据占用内存很大，所以对于数据量比较大的数据，要用生成器表达式
g = (i for i in range(1000000))
print(g)
print(g.__next__)
print(g.__next__)

扩展(字典推导式、集合推导式)

字典推导式

1.将一个字典的key和value对调。

mcase = {'a': 10, 'b': 34}
mcase_frequency = {mcase[k]: k for k in mcase}
print(mcase_frequency )
# {10: 'a', 34: 'b'}

2.合并大小写对应的value值，将k统一成小写。

mcase = {'a': 10, 'b': 34, 'A':7, 'Z':3}
mcase_frequency = {k.lower(): mcase.get(k.lower(), 0) + mcase.get(k.upper(), 0) for k in mcase.keys()}
print(mcase_frequency)
# {'a': 17, 'b': 34, 'z': 3}

集合推导式

计算列表中每个值的平方，自带去重功能。

squared = {x**2 for x in [1, -1, 2]}
print(squared)
# {1, 4}