主页 / coding / 问题 / 76931867
Accepted
AaronJPung
Asked: 2023-08-19 02:42:14 +0800 CST

检查列表数组中每个元素之间的交集

  • 772

在 Python 中，我有一个不同大小的数组的列表。我想知道列表中包含的所有数组的交集。例如：

import numpy as np

array0 = [0,1,2,3,4,5,6]
array1 = [0,2,3,5,6,7,8,9,10]
array2 = [4,5,6]
array3 = [5,6,7,8,9,10,11,12,13]

array_list = [array0, array1, array2, array3]

np.intersect1d我想使用类似跨所有数组的函数array_list并获取所有数据数组的交集。快速浏览一下，答案是

intersection = np.array([5,6])

我尝试过使用列表理解，例如：

np.intersect1d([arraylist[x] for x in np.arange(len(arraylist))]),

但 Numpy 需要 2 个参数，而不是我提供的参数。出于类似的原因，

list(filter(lambda x:x in [x for x in arraylist]))

不起作用——filter需要 2 个参数，但只得到一个。

python

1 个回答

  1. Best Answer

    您可以使用 reduce() fromfunctools模块与多个列表相交np.intersect1d：

    import numpy as np
from functools import reduce

array0 = [0,1,2,3,4,5,6]
array1 = [0,2,3,5,6,7,8,9,10]
array2 = [4,5,6]
array3 = [5,6,7,8,9,10,11,12,13]

array_list = [array0, array1, array2, array3]

print(reduce(np.intersect1d,[array0, array1, array2, array3] ))

    输出：-

    [5 6]

    或用于sets将它们相交

    import numpy as np

array0 = [0,1,2,3,4,5,6]
array1 = [0,2,3,5,6,7,8,9,10]
array2 = [4,5,6]
array3 = [5,6,7,8,9,10,11,12,13]
    
array_list = [array0, array1, array2, array3]

print(set.intersection(*map(set,array_list)))

    输出：-

    {5,6}
    • 2

相关问题