Python计算两个数组(列表)的交集、差集和并集

2022-03-14 00:00:00 数组 两个 交集

Python可以很容易对两个数组(list)计算交集、差集和并集,这里需要先将数组(list)转换成集合(set),转换后对集合进行相关计算即可,皮蛋编程(pidancode.com)对这几个函数进行了总结,可以收藏后调用。

1、获取两个数组的交集

def get_intersection(a, b):
"""
   获取两个数组的交集
   :param a: 数组a
   :param b: 数组b
   :return: 返回交集
"""
return list(set(a).intersection(set(b)))

2、获取两个数组的差集:

def get_difference(a, b):
   """
   获取两个数组的差集,不区分数组a和b的长度
   :param a: 数组a
   :param b: 数组b
   :return: 返回差集
   """
   if len(a) > len(b):
      a, b = b, a
return list(set(b).difference(set(a)))

3、获取两个数组的并集:

def get_union(a, b):
    """
   获取两个数组的并集
   :param a: 数组a
   :param b: 数组b
   :return: 返回并集
   """
return list(set(a).union(set(b)))

相关文章