python 比较两个列表的重合度|极客笔记

python 比较两个列表的重合度

在数据处理和分析中，经常会遇到需要比较两个列表之间的重合度的情况。重合度即两个列表中相同元素的比例或数量。在Python中，我们可以通过多种方式来比较两个列表的重合度，本文将介绍其中的几种方法并附上示例代码。

方法一：使用集合操作

Python中的集合（Set）可以很方便地进行元素的交集、并集等操作，利用集合操作可以快速计算出两个列表的重合度。以下是使用集合操作比较两个列表重合度的示例代码：

list1 = [1, 2, 3, 4, 5]
list2 = [3, 4, 5, 6, 7]

set1 = set(list1)
set2 = set(list2)

intersection = set1 & set2
overlap_ratio = len(intersection) / min(len(set1), len(set2))
print(overlap_ratio)

上面的代码中，我们首先将两个列表转换为集合，然后利用集合的交集操作&找出两个列表中相同的元素，最后计算重合度。运行以上代码，将输出两个列表的重合度。

方法二：使用Counter计数器

Python中的collections模块提供了Counter类，可以用来统计列表中元素出现的次数，利用Counter可以很方便地计算出两个列表的重合度。以下是使用Counter计数器比较两个列表重合度的示例代码：

from collections import Counter

list1 = [1, 2, 3, 4, 5, 5]
list2 = [3, 4, 5, 5, 6, 7]

counter1 = Counter(list1)
counter2 = Counter(list2)

intersection = counter1 & counter2
overlap_ratio = sum(intersection.values()) / min(sum(counter1.values()), sum(counter2.values()))
print(overlap_ratio)

上面的代码中，我们首先使用Counter类统计两个列表中元素的出现次数，然后利用Counter的交集&找出重合的元素，最后计算重合度。运行以上代码，将输出两个列表的重合度。

方法三：使用numpy库

如果两个列表比较大且需要进行更复杂的数学运算，可以使用Python中的numpy库来进行处理。以下是使用numpy计算两个列表重合度的示例代码：

import numpy as np

list1 = [1, 2, 3, 4, 5]
list2 = [3, 4, 5, 6, 7]

array1 = np.array(list1)
array2 = np.array(list2)

intersection = len(np.intersect1d(array1, array2))
overlap_ratio = intersection / min(len(array1), len(array2))
print(overlap_ratio)

上面的代码中，我们首先将两个列表转换为numpy数组，然后使用numpy的intersect1d函数找出两个数组中相同的元素，最后计算重合度。运行以上代码，将输出两个列表的重合度。