【Python基础】Python列表生成式
编译 |VK
来源 | Towards Data Science
列表生成式是一种基于其他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亿个元素）时，应该避免列表生成式。它可能会导致你的计算机崩溃，由于内存需求量过大。
对于这样大的列表，一个更好的选择是使用一个生成器，它实际上不会在内存中创建一个大的数据结构。生成器在使用项时创建它们。用完后，生成器会将它们扔掉。生成器不会导致内存问题，但它们比列表生成式相对较慢。
谢谢你的阅读。
往期精彩回顾
本站qq群851320808，加入微信群请扫码：