查找两个列表之间的交集大小

2024-10-06 12:33:45 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图编写一个函数,返回两个列表之间的公共元素数。如果元素在L1中出现j次,在L2中出现k次,则共享jk个元素的最小值

示例:

L1 = [1, 2, 3, 4, 5]
L2 = [4, 2]
L3 = [1, 2, 3, 4, 4, 5, 5]
intersection_size ( L1 , L2 ) => 2
intersection_size ( L1 , L3 ) => 5

我的想法是按升序对L1L2进行排序,然后比较每个元素:

def intersection(L1, L2):
    dL1 = L1[:]
    dL2 = L2[:]
    dL1.sort()
    dL2.sort()
    if dL1[1:] == [] or dL2[1:] == []:
        return 0
    if dL1[0] == dL2[0]:
        return 1 + intersection(dL1[1:], dL2[1:])
    elif dL1[0] > dL2[0]:
        return 0 + intersection(dL1, dL2[1:])
    elif dL1[0] < dL2[0]:
        return 0 + intersection(dL1[1:], dL2)

但是,当我测试intersection([1, 6, 1, 4], [1, 2, 3, 4])时,函数给我的是1而不是2。有人能告诉我哪部分错了吗

我创建了dL1dL2,因为我希望L1L2在其他方面保持不变,而且因为我是初学者,我只允许使用基本的递归和抽象列表函数来实现这一点


Tags: 函数元素l1示例列表sizereturnif
2条回答

这一部分在逻辑上不正确:

    if dL1[1:] == [] or dL2[1:] == []:
        return 0

基本情况应该是一个或两个列表都为空。但当一个或两个列表都有一个元素时,返回0。这是不正确的,因为该元素仍然可能是匹配的,所以交叉点的大小不会是0。如果将此条件更改为dL1 == [] or dL2 == [],则它可以工作:

>>> intersection(L1, L2)
2
>>> intersection(L1, L3)
5
>>> intersection([1, 6, 1, 4], [1, 2, 3, 4])
2
>>> intersection([1, 1, 1, 2, 2], [1, 1, 2, 2, 2])
4

以下方法将起作用

len(set(L1) & set(L3))

如果不想使用sets,可以执行以下操作:

def intersection(L1, L2):
    return len([x for x in L1 if x in L2])

相关问题 更多 >