得到numpy矩阵行与数组的交集

时间:2018-01-15 15:49:46

标签: python numpy numpy-ufunc

我有很多int32 numpy 1d数组。有没有一种有效的方法来获取与测试数组具有非空交集的行的索引?基本上我正在寻找更快的替代pythonic表达式:

indexes = [
    i for i, n in enumerate(iterable_of_iterables)  # [[1, 2], [2, 3], [4]]
    if set(n).intersection(test_iterable)           # [1, 4]
]                                                   # -> [0, 2]

0 个答案:

没有答案