Python轻量级循环-列表推导式

开发 后端
Python 列表推导式(list comprehension)利用其他列表创建新的列表,工作方式类似于for循环,使用列表推导式可以快速生成一个列表,或者根据某个列表生成满足指定需求的列表。

概 述

Python 列表推导式(list comprehension)利用其他列表创建新的列表,工作方式类似于for循环,使用列表推导式可以快速生成一个列表,或者根据某个列表生成满足指定需求的列表。

单循环

  1. [i for i in range(10)] 
  2. [0, 1, 2, 3, 4, 5, 6, 7, 8, 9] 

单循环+条件

0-20中所有能被3整除的数是多少

  1. [i for i in range(20) if i%3==0] 
  2. [0, 3, 6, 9, 12, 15, 18] 

多循环-2次

'ABC'和'EGF'所有可能的两两组合

  1. [i+j for i in 'ABC' for j in 'EFG'] 
  2. ['AE', 'AF', 'AG', 'BE', 'BF', 'BG', 'CE', 'CF', 'CG'] 

多循环-3次

'ABC'、'EFG'、‘HIJ’所有可能的三三组合

  1. [i+j+u for i in 'ABC' for j in 'EFG' for u in 'HIJ'] 
  2. ['AEH', 'AEI', 'AEJ', 'AFH', 'AFI', 'AFJ', 'AGH', 'AGI', 'AGJ',  
  3. 'BEH', 'BEI', 'BEJ', 'BFH', 'BFI', 'BFJ', 'BGH', 'BGI', 'BGJ', 
  4.  'CEH', 'CEI', 'CEJ', 'CFH', 'CFI', 'CFJ', 'CGH', 'CGI', 'CGJ'] 

多循环+单条件

首字母相同的男孩 女孩组合:

 

  1. girls = ['alice','bernice','clarice'] 
  2. boys  = ['chris','arnold','bob','bernod'] 
  3. [i+'<->'+j for i in girls for j in boys if i[0]==j[0]] 
  4. ['alice<->arnold', 'bernice<->bob', 'bernice<->bernod',  
  5. 'clarice<->chris'] 

多循环+多条件

第一个字母+第二个字母均相同

  1. [i+'<->'+j for i in girls for j in boys if (i[0]==j[0] and i[1]==j[1])] 
  2.  ['bernice<->bernod'] 

元组循环

列表可以直接循环,改成元组也是可以的,但是直接生产了迭代器:

  1. (i+'<->'+j for i in girls for j in boys if i[0]==j[0]) 
  2. <generator object <genexpr> at 0x0000015413B781C8> 
  3.  
  4. tuples = (i+'<->'+j for i in girls for j in boys if i[0]==j[0]) 
  5. for t in tuples: 
  6.     print(t) 
  7. alice<->arnold 
  8. bernice<->bob 
  9. bernice<->bernod 
  10. clarice<->chris 

效率问题

男孩-女孩名字对的例子中,其实效率不是很高,因为程序会检测每个可能的配对。Python中有很多解决这个问题的方法,下面是其中一种比较高效的方法。

  1. letterGirls = {} 
  2. for girl in girls: 
  3.     letterGirls.setdefault(girl[0],[]).append(girl) 
  4. print([i+'<->'+j for j in boys  for i in letterGirls[j[0]]]) 
  5. ['clarice<->chris', 'alice<->arnold', 'bernice<->bob',  
  6. 'bernice<->bernod'] 

看看我们构建的字典长啥样

  1. letterGirls 
  2. {'a': ['alice'], 'b': ['bernice'], 'c': ['clarice']} 

注意:setdefault()方法,是字典的一个方法,类似于get()方法,如果键不存在于字典中,将会添加键并将值设为默认值。如果存在,则返回该字段的值。

语法:

  1. dict.setdefault(key, default=None

 

责任编辑:赵宁宁 来源: AI入门学习
相关推荐

2024-05-20 12:00:00

Python列表推导式

2024-05-20 08:30:00

Python编程

2021-03-03 12:55:30

Python列表推导式代码

2024-02-21 20:43:02

Python列表推导式

2023-04-28 07:42:02

2022-08-10 12:21:07

PythonWebBottle

2022-07-15 16:39:19

PythonWhoosh工具

2024-06-21 15:19:40

2024-10-28 17:06:50

2023-11-24 11:11:08

Python数据库

2009-07-17 14:38:51

轻量级Swing组件

2009-07-14 18:05:28

轻量级Swing组件

2015-06-17 14:10:34

Redis分布式系统协调

2024-04-29 08:42:23

2023-06-13 13:38:00

FlaskPython

2023-12-18 10:24:59

2023-08-09 08:01:38

场景Redis接口

2009-09-11 08:26:49

Linux系统CRUX 2.6Linux

2016-10-14 16:35:39

2023-12-07 19:33:09

Python推导式
点赞
收藏

51CTO技术栈公众号