未知列表v长度时，如何实现各numpy数组指定列的集合交集？

当然可以，不管列表v的长度是多少，都能实现所有数组第一列元素的集合交集运算。可以借助functools.reduce来简化多集合的迭代交集操作，具体实现如下：

代码示例 #

import numpy as np
from functools import reduce

# 示例：v可包含任意数量的numpy数组
v = [
    np.array([(4,'3'),(5,6)]),
    np.array([(4,'a'),(5,'b')]),
    np.array([(4,'x'),(6,'y')])
]

# 提取每个数组第一列并转为集合
column_sets = [set(arr[:, 0]) for arr in v]
# 计算所有集合的交集
common_elements = reduce(lambda a, b: a & b, column_sets)

print(common_elements)  # 输出: {4}

说明 #

• 通过列表推导式遍历v，批量提取每个数组的第一列并转换为集合，避免手动逐个处理。
• reduce函数会依次将所有集合做交集运算（a & b等价于集合的intersection方法），最终得到所有数组第一列的公共元素。
• 该写法完全不依赖v的长度，哪怕v只有单个元素，也能正常返回对应数组第一列的集合。

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