(Numpy) 索引列表到布尔数组

Question

输入：

1. 数组长度（整数）
2. 索引（集合或列表）

输出：

一个布尔 numpy 数组，其索引值为 1，其他索引值为 0。

---

例子：

输入： array_length=10, indexes={2,5,6}

输出：

[0,0,1,0,0,1,1,0,0,0]

这是我的一个简单实现：

def indexes2booleanvec(size, indexes):
    v = numpy.zeros(size)
    for index in indexes:
        v[index] = 1.0
    return v

有没有更优雅的方法来实现这个？

Answer 1

一种方法是避免循环

In [7]: fill = np.zeros(array_length)     #  array_length = 10

In [8]: fill[indexes] = 1                 #  indexes = [2,5,6]

In [9]: fill
Out[9]: array([ 0.,  0.,  1.,  0.,  0.,  1.,  1.,  0.,  0.,  0.])

Answer 2

另一种方法（在一行中）：

np.isin(np.arange(array_length), indexes)

然而，这比 Zero 的解决方案慢。