Python计算两个数组(列表)的交集、差集和并集
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)))
相关文章