python的列表生成式
列表生成式是一种基于其他iterable(如集合、元组、其他列表等)创建列表的方法。它还可以用更简单、更吸引人的语法表示for和if循环。不过,列表生成式比for循环要快得多。
列表生成式的基本结构如下:
这看起来很简单,但在某些情况下可能会变得棘手。在本文中,我们将从一个非常简单的列表生成式开始,并逐步增加复杂性。我将清楚地解释如何表示和生成式高度复杂的列表生成式。
在大多数情况下,列表生成式优先于for和if循环,因为:
- 它们比for循环快得多
- 它们被认为比循环和映射函数更具python特性
- 列表生成式的语法更容易阅读
我们从一个简单的例子开始。
words = ['data','science','machine','learning']
我们要创建一个列表,其中包含单词列表中每个单词的长度。我们使用for循环和列表生成式来执行任务。
# for循环a = []
for word in words:
a.append(len(word))
# 列表生成式
b = [len(word) for word in words]
print(f"a is {a}")
print(f"b is {b}")
a is [4, 7, 7, 8]
b is [4, 7, 7, 8]
生成式列表语法的最佳方法是比较for和if循环。下图演示了比较:
for循环末尾的表达式放在列表的开头。
当我们复习这些例子时,会更清楚。让我们创建一个列表,其中包含单词列表中长度大于5的项。
# for循环a = []
for word in words:
if len(word) > 5:
a.append(word)
# 列表生成式
b = [word for word in words if len(word) > 5]
print(f"a is {a}")
print(f"b is {b}")
a is ['science', 'machine', 'learning']
b is ['science', 'machine', 'learning']
下图显示了如何在列表中表示循环中的表达式。
与上一个示例一样,在for循环末尾所做的事情被放在列表的开头。在这种情况下,我们按原样处理项。列表生成式的下一部分是循环中的第一个表达式,然后是循环中的第二个表达式。
循环中的逻辑更直观。因此,与循环相比,列表生成式的结构相对容易掌握。过一段时间,你就不必做一个明确的比较,因为你的大脑已经习惯了。
下一个例子稍微复杂一些。我们要创建一个包含单词列表项中所有“a”、“e”和“i”字母的列表。此任务涉及嵌套的for和if循环。
# for循环a = []
for word in words:
for letter in word:
if letter in ["a","e","i"]:
a.append(letter)
详细说明一下语法:反复浏览单词表。对于每个项,遍历字符。如果一个字符符合给定的条件,它将被附加到列表a中。
完成相同任务的列表生成式如下。
b = [letter for word in words for letter in word if letter in ["a","e","i"]]
我们从for循环中的最后一个表达式开始生成式列表。剩下的部分从循环的开头开始。下图说明了每个步骤。
结论
列表生成式是一种高效的操作。对于简单的任务,生成式的语法很容易,在复杂的情况下可能会变得棘手。
如果你很难创建或生成式复杂的列表生成式,请尝试使用循环编写。
需要注意的是,列表生成式并不总是最佳选择。它们将整个输出列表加载到内存中。这对于中小型列表是可以接受的,甚至是可取的,因为它使操作更快。
然而,当我们处理大型列表(例如10亿个元素)时,应该避免列表生成式。它可能会导致你的计算机崩溃,由于内存需求量过大。
对于这样大的列表,一个更好的选择是使用一个生成器,它实际上不会在内存中创建一个大的数据结构。生成器在使用项时创建它们。用完后,生成器会将它们扔掉。生成器不会导致内存问题,但它们比列表生成式相对较慢。
转载自https://www.zhihu.com/people/panchuangai
以上是 python的列表生成式 的全部内容, 来源链接: utcz.com/z/386851.html