Python列表推导式

以你之姓@ 2023-10-03 09:17 18阅读 0赞

列表推导式提供了一种创建列表的简洁方法。

它由包含表达式的括号组成，后跟一个 for 子句，然后是  
零个或多个 for 或 if 子句。表达式可以是任何东西，这意味着您可以  
将各种对象放入列表中。

结果将是一个新列表，该列表是通过在其  
后面的 for 和 if 子句的上下文中评估表达式而产生的。

列表推导式总是返回一个结果列表。

如果你曾经这样做过：

new_list = []
    for i in old_list:
        if filter(i):
            new_list.append(expressions(i))

您可以使用列表理解获得相同的结果。注意 append 方法已经消失了！

new_list = [expression(i) for i in old_list if filter(i)]

### 句法 ###

列表推导式以**方括号**' **\[' 和 '\]'**开头，以帮助您记住  
结果将是一个列表。

该[基本语法][Link 1]使用方括号。

[ expression for item in list if conditional ]

这相当于：

for item in list:
        if conditional:
            expression

让我们分解一下，看看它有什么作用。

new_list = [expression(i) for i in old_list if filter(i)]

**new\_list**  
新列表（结果）。

**expression(i)**  
表达式基于用于旧列表中每个元素的变量。

**for i in old\_list**  
单词 for 后跟要使用的变量名，后跟  
旧列表中的单词。

**if filter(i)**  
应用带有 If 语句的过滤器。

此[博客][Link 2]展示了如何直观地分解列表理解的示例：

**new\_range = \[i \* i for i in range(5) if i % 2 == 0\]**  
  
对应于：

**\*结果\* = \[\*变换\* \*迭代\* \*过滤器\* \]**

\* 运算符用于重复。过滤器部分回答了是否  
应该转换项目的问题。

### 例子 ###

现在当我们知道列表推导式的语法后，让我们展示一些示例以及  
如何使用它。

#### 创建一个简单的列表 ####

让我们从创建一个简单的列表开始。

x = [i for i in range(10)]
    print (x)
    
    # This will give the output:
    [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

这样您就可以创建一个简单的列表。

#### 使用循环和列表推导式创建列表 ####

对于下一个示例，假设我们要创建一个平方列表。从一个空列表开始。

# You can either use loops:
    squares = []
    
    for x in range(10):
        squares.append(x**2)
     
    print (squares)
    [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
    
    # Or you can use list comprehensions to get the same result:
    squares = [x**2 for x in range(10)]
    
    print (squares)
    [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

只需记住语法：\[如果有条件，则列表中的项目的表达式\]

#### 将列表的部分相乘 ####

将列表的每个部分乘以 3 并将其分配给一个新列表。

list1 = [3,4,5]
     
    multiplied = [item*3 for item in list1] 
     
    print (multiplied)
    [9,12,15]

请注意 item\*3 如何将每件乘以 3。

#### 显示每个单词的第一个字母 ####

我们将取每个单词的第一个字母并从中列出一个列表。

listOfWords = ["this","is","a","list","of","words"]
    
    items = [ word[0] for word in listOfWords ]
    
    print (items)

输出应该是：\['t', 'i', 'a', 'l', 'o', 'w'\]

#### 小写/大写转换器 ####

让我们展示转换小写/大写字母是多么容易。

>>> [x.lower() for x in ["A","B","C"]]
    ['a', 'b', 'c']
    
    >>> [x.upper() for x in ["a","b","c"]]
    ['A', 'B', 'C']

#### 仅从给定字符串打印数字 ####

此示例显示如何从字符串中提取所有数字。

string = "Hello 12345 World"
    numbers = [x for x in string if x.isdigit()]
    print (numbers)
    
    >> ['1', '2', '3', '4', '5']

如果您不想要任何数字，请将 x.isdigit() 更改为 x.isalpha()。

#### 使用列表理解解析文件 ####

在这个例子中，我们可以看到如何从文本文件中取出特定的行。

创建一个文本文件并在其中放入一些文本。

this is line1  
this is line2  
this is line3  
this is line4  
this is line5

将文件另存为 test.txt

# Then create the filter by using list comprehension:
    
    fh = open("test.txt", "r")
    
    result = [i for i in fh if "line3" in i]
    
    print (result)

输出：\['this is line3\\n'\]

[Link 1]: https://www.pythonforbeginners.com/basics/python-syntax-basics
[Link 2]: http://blog.cdleary.com/2010/04/learning-python-by-example-list-comprehensions/