Python推导式|极客笔记

Python推导式

在本文中，我们将看到如何在Python的数据结构中使用推导式，如列表、字典、集合和生成器。

推导式提供了一种精确的方法来编写Python程序。它减小了代码的长度，同时不影响其易读性。

因此，我们将讨论以下推导式：

列表推导式
字典推导式
集合推导式
生成器推导式

列表推导式

我们知道，列表的元素被包含在方括号中，它可以保存多个数据类型的值。

在下面的程序中，我们将从列表中取出偶数。

让我们看一下下面的列表示例。

示例

list1= [20,25,24,30,35,40,44]
list2=[]
for i in list1:
    if i%2==0:
        list2.append(i)
print("The elements of list2 are :" ,list2)

输出 –

The elements of list2 are : [20, 24, 30, 40, 44]

在这里，我们指定了list1的元素，然后使用for循环来取出每个元素，并使用模运算符来检查它是否可被2整除。

在下面给出的程序中，我们使用列表推导式来执行相同的操作。

示例 2-使用列表推导式

list1=[20,25,24,30,35,40,44]
list2=[i for i in list1 if i%2==0]
print("The elements obtained using list comprehension are :" ,list2)

输出-

The elements obtained using list comprehension are : [20, 24, 30, 40, 44]

在此，我们可以观察到我们在list2中提供了for循环和条件判断的理解。

下一个程序是基于获取list1中所有元素的立方。

示例3

list1=[2,3,4,5,6,7,8,9,10]
list2=[]
for i in list1:
    list2.append(i**3)
print("The cube of the elements present in list1 is: ",list2)

输出-

The cube of the elements present in list1 is:  [8, 27, 64, 125, 216, 343, 512, 729, 1000]

我们使用append()方法，将list2中每个元素的立方保存起来，然后显示出来。

我们可以使用列表推导式来实现同样的功能-

示例4 使用列表推导式

list1=[2,3,4,5,6,7,8,9,10]
list2=[i**3 for i in list1 ]
print("The cube of the elements obtained by list comprehension is: ",list2)

输出-

The cube of the elements obtained by list comprehension is:  [8, 27, 64, 125, 216, 343, 512, 729, 1000]

字典推导

我们都知道字典使用键值对，让我们来看看显示这些键值对的程序。

示例

fruits=['Apple','Bananas','Custard Apple','Pineapple','Blueberries']
color=['Red','Yellow','Green','Brown','Violet']
result_dict={}
for key,value in zip(fruits,color):
    result_dict[key]=value
print("The resultant dictionary would be: ",result_dict)

输出：

The resultant dictionary would be:  {'Apple': 'Red', 'Bananas': 'Yellow', 'Custard Apple': 'Green', 'Pineapple': 'Brown', 'Blueberries': 'Violet'}

在下一个程序中，使用推导来实现相同的事情

示例2 使用字典推导

fruits=['Apple','Bananas','Custard Apple','Pineapple','Blueberries']
color=['Red','Yellow','Green','Brown','Violet']
result_dict={key:value for (key,value) in zip(fruits,color)}
print("The resultant dictionary using comprehension would be: ",result_dict)

输出-

The resultant dictionary using comprehension would be:  {'Apple': 'Red', 'Bananas': 'Yellow', 'Custard Apple': 'Green', 'Pineapple': 'Brown', 'Blueberries': 'Violet'}

在这里，我们可以看到我们在 result_dict 中提供了理解，我们在其中提供了显示来自水果和颜色列表的键值对的表达式。

集合推导

集合用于显示给定集合中的唯一元素。让我们使用集合来获得列表中所有元素的平方。

示例1

list1=[2,3,4,5,6,7,8,9,10]
result_set=set()
for i in list1:
    result_set.add(i**2)
print("The square of the numbers present in list1 is: ",result_set)

输出-

The square of the numbers present in list1 is:  {64, 4, 36, 100, 9, 16, 49, 81, 25}

在下面的程序中，我们使用推导完成了相同的操作。

示例 2- 使用集合推导

list1=[2,3,4,5,6,7,8,9,10]
result_set={i**2 for i in list1}
print("The square of the numbers obtained through set comprehension: ",result_set)

输出-

The square of the numbers obtained through set comprehension:  {64, 4, 36, 100, 9, 16, 49, 81, 25}

我们已经从list1中获取了每个元素，并在result_set中提供了计算这些元素平方的表达式。

生成器推导式

生成器与函数非常相似。它使用yield关键字生成一个值。让我们看看在这里如何使用推导式。

示例

list1=[12,16,17,20,21,24,28,30,31]
result_gen=(i for i in list1 if i%2==0)
for i in result_gen:
    print("The element which is even in list1 is: ",i)

输出 –

The element which is even in list1 is:  12
The element which is even in list1 is:  16
The element which is even in list1 is:  20
The element which is even in list1 is:  24
The element which is even in list1 is:  28
The element which is even in list1 is:  30

执行程序后，它会显示list1中的偶数元素。