欢迎访问宙启技术站
智能推送

Python函数实例:如何计算两个列表的交集和并集

发布时间:2023-05-31 15:17:07

Python函数是一段有名字的程序代码块,它用来接收输入参数并执行一定的计算来产生输出结果。本文主要介绍Python中计算两个列表的交集和并集的函数。

计算两个列表的交集:

列表的交集指的是两个列表中共有的元素,我们可以使用Python中的set()函数来数学上求解两个集合的交集。

def intersection(list1, list2):
    set1 = set(list1)
    set2 = set(list2)
    result_set = set1.intersection(set2)
    result_list = list(result_set)
    return result_list

这个函数接收两个列表作为参数,并使用set()函数将列表转化为集合,利用集合的intersection()方法求出两个集合的交集。最后再将得到的集合转化为列表。我们可以通过下面的代码来验证函数的正确性。

list1 = [1, 2, 3, 4, 5]
list2 = [3, 4, 5, 6, 7]
print(intersection(list1, list2))

输出结果:

[3, 4, 5]

计算两个列表的并集:

列表的并集指的是两个列表中所有元素的集合,我们可以使用Python中的set()函数来数学上求解两个集合的并集。

def union(list1, list2):
    set1 = set(list1)
    set2 = set(list2)
    result_set = set1.union(set2)
    result_list = list(result_set)
    return result_list

这个函数也接收两个列表作为参数,并使用set()函数将列表转化为集合,利用集合的union()方法求出两个集合的并集。最后再将得到的集合转化为列表。我们可以通过下面的代码来验证函数的正确性。

list1 = [1, 2, 3, 4, 5]
list2 = [3, 4, 5, 6, 7]
print(union(list1, list2))

输出结果:

[1, 2, 3, 4, 5, 6, 7]

总结:

本文中介绍了Python中计算两个列表的交集和并集的函数,使用集合来数学上求解这些问题,这些函数的原理并不难懂,熟记这些函数可以帮助我们更好地处理列表问题。另外,在实际开发中,我们可以将这些函数封装到模块中并给它们起一个有意义的名称,以便更好地复用它们。