python的列表生成式

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

回到顶部