如何用Pythonic方法求两个数组列表的所有交集与差集？

针对你提出的需求——拆分两个numpy数组列表，提取出所有跨列表的交集和各自的差集（单个列表内数组无交集），我整理了一个清晰的解决方案，结合集合运算来高效处理：

思路分析 #

核心逻辑是利用集合的快速交、差运算来拆分元素：

1. 将每个numpy数组转换为集合，方便进行集合操作；
2. 先提取所有跨列表的交集部分，同时从原集合中移除这些交集元素，避免重复计算；
3. 收集每个列表中剩余的独有关联元素（差集部分）；
4. 将所有非空的集合转换回numpy数组，得到最终结果。

代码实现 #

import numpy as np

def split_arrays(x_list, y_list):
    # 将numpy数组列表转换为集合列表，便于集合运算
    x_sets = [set(arr) for arr in x_list]
    y_sets = [set(arr) for arr in y_list]
    
    results = []
    
    # 提取所有x与y的交集部分
    intersections = []
    for x_set in x_sets:
        for y_set in y_sets:
            common = x_set & y_set
            if common:
                intersections.append(common)
                # 从原集合中移除已提取的交集元素，避免重复处理
                x_set -= common
                y_set -= common
    
    # 将交集转换为numpy数组并加入结果
    for inter in intersections:
        # 排序保证结果一致性，若不需要排序可去掉sorted
        results.append(np.array(sorted(inter)))
    
    # 提取x中剩余的独有元素（差集）
    for x_set in x_sets:
        if x_set:
            results.append(np.array(sorted(x_set)))
    
    # 提取y中剩余的独有元素（差集）
    for y_set in y_sets:
        if y_set:
            results.append(np.array(sorted(y_set)))
    
    return results

# 测试示例
if __name__ == "__main__":
    x = [np.array([0, 6, 7, 10]), np.array([1, 2, 5, 9])]
    y = [np.array([7, 10]), np.array([8]), np.array([0, 3, 4, 5])]
    
    final_result = split_arrays(x, y)
    print("拆分结果：")
    for arr in final_result:
        print(arr)

输出结果 #

运行上述代码后，输出为：

[ 7 10]
[0]
[5]
[6]
[1 2 9]
[8]
[3 4]

和你预期的结果元素完全一致，只是顺序略有不同。如果需要严格匹配预期顺序，可以在结果收集后调整排序逻辑，或者在处理时按特定顺序遍历。

注意事项 #

• 由于集合是无序的，代码中使用sorted()来保证数组元素的有序性，若你的场景不需要有序，可以直接移除sorted()；
• 题目中明确单个列表内的数组无交集，因此无需处理同一列表内数组的元素重叠，这让逻辑更简洁高效；
• 集合运算的时间复杂度较低，对于大规模数组列表也能保持较好的性能。

内容的提问来源于stack exchange，提问作者Jorge