Python 列表推导式使用有哪些注意事项?(干货建议收藏)

Python 列表推导式并不是给初学者用的,因为它非常反直觉,甚至对于有其他编程语言背景的人也是如此。

我们接触到 List 的使用时,学习的内容都是零散的。所以我们缺少一个关于如何在各种各样的场景下使用 List 的知识体系。

本文提供了一些 List 的使用指南,尽可能涵盖各个方面。希望本文可以成为你的一站式实用手册。

如果大家在学习中遇到困难,想找一个python学习交流环境,可以加入我们的python裙,裙号930900780,可领取python学习资料,会节约很多时间,减少很多遇到的难题。

使用建议

1.建议使用迭代的方式

使用 List 最基本的方式是以一个可迭代对象为基础,创建一个 List 对象,这个可迭代对象可以是任意可以迭代元素的Python对象。使用方法如下。

[expression for item in iterable]

复制代码

下面这段代码展示了一个使用列表相关技术创建 List 对象的例子。在这个例子中,我们定义了一个 Integer 列表,并基于这个对象创建了保存每个数字的平方数和立方数的 List 对象。

>>> # 创建一个 Integer 列表

>>> integers = [1, 2, 3, 4, 5, 6]

>>> # 创建平方数和立方数列表

>>> powers = [(x*x, pow(x, 3)) for x in integers]

>>> print(powers)

[(1, 1), (4, 8), (9, 27), (16, 64), (25, 125), (36, 216)]

复制代码

上面的例子把 List 对象当作迭代器使用。我们应该知道,许多类型的对象也是可迭代的,比如 List、Set、Dictionary 和 String 等等。其他数据类型,像 range、map、filter,以及 pandas 包中的 Series、DataFrame,都是可迭代的。下面的代码演示了某些对象的使用方法。

>>> # 使用 range 对象

>>> integer_range = range(5)

>>> [x*x for x in integer_range]

[0, 1, 4, 9, 16]

>>> # 使用 Series 对象

>>> import pandas as pd

>>> pd_series = pd.Series(range(5))

>>> print(pd_series)

0    0

1    1

2    2

3    3

4    4

dtype: int64

>>> [x*x for x in pd_series]

[0, 1, 4, 9, 16]

复制代码

2.如果只需用到其中的某些元素,应当使用条件判断语句

假设你需要将符合某种条件的元素归集起来,并创建一个 list。下面展示了相关的语法。

[expression for item in iterable if condition]

复制代码

if 语句用来实现条件判断。下面的代码展示了这种用法的一个简单示例。

>>> # 同样创建一个 Integer 列表

>>> integers = [1, 2, 3, 4, 5, 6]

>>> # 筛选出偶数,创建一个这些偶数的平方数列表

>>> squares_of_evens = [x*x for x in integers if x % 2 == 0]

>>> print((squares_of_evens))

[4, 16, 36]

复制代码

3.使用条件判断语句

List 对象中还可以使用 if-else 形式的条件判断,语法如下。

[expression0 if condition else expression1 for item in iterable]

复制代码

这跟前面的那种用法有些类似,别把这两种用法混淆。在本例中,条件语句本身是一个整体。下面的代码提供了一个例子。

>>> # 创建一个 Integer 列表

>>> integers = [1, 2, 3, 4, 5, 6]

>>> # 遍历 integers 中的元素,如果是偶数,取平方数存入新的列表

>>> # 如果是奇数,取立方数存入新的列表

>>> custom_powers = [x*x if x % 2 == 0 else pow(x, 3) for x in integers]

>>> print(custom_powers)

[1, 4, 27, 16, 125, 36]

复制代码

4.如果有嵌套结构,可以使用嵌套的循环

有可能可迭代对象中的元素自身也是可迭代的,尽管这种情况不太常见。如果你对嵌套的可迭代对象有兴趣,可以使用 for 来实现循环嵌套。语法如下。

[expression for item_outer in iterable for item_inner in item_outer]

# 与下面的代码等同

for item_outer in iterable:

    for item_inner in item_outer:

        expression

复制代码

上面的代码展示了使用for实现嵌套循环的例子。

>>> # 创建一个包含元组的列表

>>> prices = [('$5.99', '$4.99'), ('$3.5', '$4.5')]

>>> # 获取元组中的每个价格,以此创建一个一维列表

>>> prices_formatted = [float(x[1:]) for price_group in prices for x in price_group]

>>> print(prices_formatted)

[5.99, 4.99, 3.5, 4.5]

复制代码

5.替换高阶函数

有的人比较习惯函数式编程,比如使用高阶函数也是这种习惯的表现之一。特别说明一下,高阶函数是那些需要使用输入或输出参数的函数。在 Python 中,常用的高阶函数有 map() 和 filter()。

>>> # 创建一个 integer 类型的列表

>>> integers = [1, 2, 3, 4, 5]

>>> # 使用 map 创建平方数列表

>>> squares_mapped = list(map(lambda x: x*x, integers))

>>> squares_mapped

[1, 4, 9, 16, 25]

>>> # 使用列表推导式创建平方数列表

>>> squares_listcomp = [x*x for x in integers]

>>> squares_listcomp

[1, 4, 9, 16, 25]

>>> # 使用 filter 取得 integers 中的偶数列表

>>> filtered_filter = list(filter(lambda x: x % 2 == 0, integers))

>>> filtered_filter

[2, 4]

>>> # 使用列表推导式取得 integers 中的偶数列表

>>> filterd_listcomp = [x for x in integers if x % 2 == 0]

>>> filterd_listcomp

[2, 4]

复制代码

从上面的例子可以看出,使用 list 的某些特性比使用高阶函数更具有可读性,而且也能实现较复杂的嵌套结构。

使用禁忌

1.不要忘了定义构造函数

有人认为列表推导式很酷炫,是 Python 特有的功能,所以为了炫耀自己的 Python 水平,即使有更好替代方案也要使用它。

>>> # 使用 range 创建列表对象

>>> numbers = [x for x in range(5)]

>>> print(numbers)

[0, 1, 2, 3, 4]

>>> # 以一个字符串为基础,创建一个小写字母的字符列表

>>> letters = [x.lower() for x in 'Smith']

>>> print(letters)

['s', 'm', 'i', 't', 'h']

复制代码

上述例子中,我们使用了 range 和 string,这两种数据结构都是可迭代的,list()构造函数可以直接使用 iterable 创建一个 list 对象。下面的代码提供了更合理的解决方案。

>>> # 使用 range 创建列表对象

>>> numbers = list(range(5))

>>> print(numbers)

[0, 1, 2, 3, 4]

>>> # 以一个字符串为基础,创建一个小写字母的字符列表

>>> letters = list('Smith'.lower())

>>> print(letters)

['s', 'm', 'i', 't', 'h']

复制代码

2.不要忘了生成器表达式

在 Python 中,生成器是一种特殊的可迭代对象,它会延迟加载元素,直到被请求才会加载。这在处理大量数据时会非常高效,它能提升存储效率。相比之下,list 对象为了方便计数和索引,一次性创建所有的元素。所以跟生成器相比,在元素个数相同时,list 需要占用更多内存。

我们可以定义一个生成器函数来创建生成器。我们也可以使用下面的语句来创建生成器,这是一种称为生成器表达式的方法。

(expression for item in iterable)

复制代码

你可能会注意到,除了使用圆括号外,它的语法跟使用 list 的语句很相似。所以需要注意区分。

考虑下面这个例子。我们要计算前一百万个数字的平方和。如果使用 list 来实现,方法如下。

>>> # 创建列表对象 squares

>>> squares = [x*x for x in range(10_000_000)]

>>> # 计算它们的总和

>>> sum(squares)

333333283333335000000

>>> squares.__sizeof__()

81528032

复制代码

如上所示,list 对象占据 81528032 字节。我们考虑使用 generator 进行相同的操作,代码如下。

>>> # 创建 generator 对象,保存每个数的平方数

>>> squares_gen = (x*x for x in range(10_000_000))

>>> # 计算它们的总和

>>> sum(squares_gen)

333333283333335000000

>>> squares_gen.__sizeof__()

96

复制代码

跟使用 list 相比,使用 generator 内存开销小得多,只有 96 字节。原因很简单———— generator 不需要获取所有的元素。相反,它只需要获取各个元素在序列中的位置,创建下一个元素并呈现它,而且不必保存在内存中。

结论

本文中,我们整理了 list 应用的一些关键要领。这些该做的和不该做的都非常清晰明了。我估计你会在合适的场景中用到它。下面是本文内容的小结。

使用迭代的方式。Python 中有许多类型的 iterable,你应当在掌握基础(list 和 tuple)的同时融会贯通。

使用条件判断语句。如果你对在 iterable 中筛选某些元素感兴趣,可以多多研究条件判断。

使用条件判断表达式。如果你需要有选择性地获取某些数据,可以使用条件判断表达式。

使用嵌套的循环。如果你要处理嵌套的 iterable,可以使用嵌套的循环结构。

用 list 替代高阶函数在很多情况下,可以用 list 替代高阶函数。

不要忘记 list 的构造函数定义 list 的构造函数,可以使用 iterable 创建一个 list 对象。如果你直接使用 iterable,推荐用这个方法。

不要忘了生成器表达式它的语法与 list 中的语法相似。在处理大量的对象时,这是一种节省内存开销的办法。list 和 generator 不同的是,为了日后的索引和访问, list 必须提前创建,如果元素个数很多,就会消耗很大的内存。

最后多说一句,想学习Python可联系小编,这里有我自己整理的整套python学习资料和路线,想要这些资料的都可以进q裙930900780领取。

本文章素材来源于网络,如有侵权请联系删除。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,684评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,143评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,214评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,788评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,796评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,665评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,027评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,679评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 41,346评论 1 299
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,664评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,766评论 1 331
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,412评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,015评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,974评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,203评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,073评论 2 350
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,501评论 2 343