在编写程序或者查看别人的程序时，经常会遇到列表生成式，这个使用起来并不复杂，但是非常有用，使我们的代码更加简洁灵活。很多python使用者并不太会使用它。今天，就给大家详细讲解列表生成式和生成器表达式的使用

 

优点：一行代码几乎可以搞定所需要的任何列表

缺点：容易着迷，不易排错，不能超过三个循环

  • 用列表推导式能构建的任何列表，用别的都可以构建，比如for循环
  • 列表推导式，最多不超过3个for循环。判断只能用一个

 

使用for循环方法

  1. li = []
  2. for i in range(1,4):
  3.     li.append('还珠格格第'+str(i)+'部')
  5. print(li)                   #['还珠格格第1部', '还珠格格第2部', '还珠格格第3部']

第二种写法

  1. li = []
  2. for i in range(1,4):
  3.     li.append('还珠格格第%s部' % i)
  5. print(li)

上面的代码，可以一行搞定。用列表推导式就可以了

  1. li = ['还珠格格第%s部' %i for i in range(1,4)]
  2. print(li)
  4. li = ['还珠格格第{}部'.format(i) for i in range(1,4)]         # 上下结果相同
  5. print(li)

例：求1～10平方结果

  1. li = [i ** 2 for i in range(1,11)]
  2. print(li) 
  1. 执行输出：[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
  1. 30以内所有能被3整除的数
  1. l3 = [i for i in range(1,31) if i % 3 == 0]
  2. print(l3)                         　　　　　　　　# [3, 6, 9, 12, 15, 18, 21, 24, 27, 30]
  1. 30以内所有能被3整除的数的平方
  1. li = [i**2 for i in range(1,31) if i % 3 == 0]
  2. print(li)                         　　　　　　　　# [9, 36, 81, 144, 225, 324, 441, 576, 729, 900]
  1. 找到嵌套列表中名字含有两个'e'的所有名字
  1. names = [['Tom', 'Billy', 'Jefferson', 'Andrew', 'Wesley', 'Steven', 'Joe'],['Alice', 'Jill', 'Ana', 'Wendy', 'Jennifer', 'Sherry']]
  2. l4 = [name for i in names for name in i if name.count('e') == 2 ]
  3. print(l4)                         　　　　　　　　# ['Jefferson', 'Wesley', 'Steven', 'Jennifer']

常用的是列表推导式

 

字典推导式

将一个字典的key和value对调

  1. dict = {'a': 10, 'b': 34}
  2. dict_frequency = {dict[k]: k for k in mcase}
  3. print(dict_frequency) 
  1. 执行输出：{10: 'a', 34: 'b'}

相当于

  1. dict = {'a': 10, 'b': 34}
  2. dict_frequency = {}
  3. for k in dict:
  4.     dict_frequency[k]=dict[k]
  6. print(dict_frequency) 
  1. 如果Key和value是一样的，不适合上面的代码

集合推导式

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

  1. squared = {x**2 for x in [1, -1, 2]}
  2. print(squared)
  1. 执行输出：{1, 4}
  1. a = {1,1,4}
  2. print(type(a)) 
  1. 执行输出：<class 'set'>　　结果是一个集合，它也是用{}表示的。

集合和字典是有区别的：

　　有键值对的，是字典，比如{‘k1′:1,’k1’:2}

　　没有键值对的，是集合，比如{1,2,3,4}

 

与列表推导式相同，只是将[ ]改为( )

  1. l_obj = ('还珠格格第%s部' % i for i in range(1,4))
  2. print(l_obj)
  1. 结果是一个生成器对象：<generator object <genexpr> at 0x000002DDBEBADE60>

取值使用__next__方法

  1. l_obj = ('还珠格格第%s部' % i for i in range(1,4))
  3. print(l_obj.__next__())
  4. print(l_obj.__next__())
  5. print(l_obj.__next__())
  1. 输出结果：
  2. 还珠格格第1部
  3. 还珠格格第2部
  4. 还珠格格第3部

列表推导式：一目了然，占内存

生成器表达式： 不便看出，节省内存。

版权声明：本文为Summer-skr--blog原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://www.cnblogs.com/Summer-skr--blog/p/11650979.html