不为成仙,只为在这红尘中等你回来。

您现在的位置是:网站首页>>Python

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

2020年4月22日 13:17 | 分类:Python | 标签: 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}