第5章:python推导式

Python 推导式是一种独特的数据处理方式,可以从一个数据序列构建另一个新的数据序列的结构体。

Python 支持各种数据结构的推导式:

  • 1)列表(list)推导式
  • 2)字典(dict)推导式
  • 3)集合(set)推导式
  • 4)元组(tuple)推导式
    推导式可以利用 range 区间、元组、列表、字典和集合等数据类型,快速生成一个满足指定需求的列表/字典/集合/元组。

1. 列表(list)推导式

列表推导式格式为:

[表达式 for 变量 in 列表]
[expression for item in Sequence ]
或者
[表达式 for 变量 in 列表 if 条件]
[expression for item in Sequence if condition]

  • expression:列表生成元素表达式,可以是有返回值的函数。
  • for item in Sequence:迭代 Sequence 将 item 传入到 expression 表达式中。
  • if condition:条件语句,可以过滤列表中不符合条件的值。

示例1:将列表names的字符串,转换成大写字母

>>> names = ['Bob','Tom','alice','Jerry','Wendy','Smith']
>>> new_names = [name.upper() for name in names]
>>> new_names
['BOB', 'TOM', 'ALICE', 'JERRY', 'WENDY', 'SMITH']

示例2:计算 30 以内可以被 3 整除的整数:

>>> ans = [x for x in range(30) if x%3==0]
>>> ans
[0, 3, 6, 9, 12, 15, 18, 21, 24, 27]

2. 字典(dict)推导式

字典推导基本格式:

{ key_expr: value_expr for value in collection }

{ key_expr: value_expr for value in collection if condition }

示例1:根据列表中各字符串及其长度创建字典

>>> listdemo = ['Google','Runoob', 'Taobao']
>>> newdict = {key:len(key) for key in listdemo}
>>> newdict
{'Google': 6, 'Runoob': 6, 'Taobao': 6}

示例2:提供三个数字,以三个数字为键,三个数字的平方为值来创建字典

>>> dict1 = {x:x**2 for x in (1,2,3)}
>>> dict1
{1: 1, 2: 4, 3: 9}

3. 集合(set)推导式

集合推导式基本格式:

{ expression for item in Sequence }

{ expression for item in Sequence if conditional }

示例1:计算数字 1,2,3 的平方数

>>> new_set = {x**2 for x in range(1,4)}
>>> new_set
{1, 4, 9}

示例2:判断不是 abc 的字母并输出

>>> a = {x for x in 'abracadabra' if x not in 'abc'}
>>> a
{'r', 'd'}

4. 元组(tuple)推导式

元组推导式返回的结果是一个生成器对象,而不是元组
元组推导式基本格式:

(expression for item in Sequence )

(expression for item in Sequence if conditional )

示例1:生成一个包含数字 1~9 的元组

>>> a = (x for x in range(1,10))
>>> a
<generator object <genexpr> at 0x7faf6ee20a50>  # 返回的是生成器对象
>>> tuple(a)       # 使用 tuple() 函数,可以直接将生成器对象转换成元组
(1, 2, 3, 4, 5, 6, 7, 8, 9)
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容