python - 获取numpy矩阵行与数组有交集

Question

我有大量 int32 numpy 一维数组。有没有一种有效的方法来获取与测试数组有非空交集的行的索引？基本上我正在寻找更快的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]