创建具有已知非零入口索引的numpy 1d布尔数组的快速方法



给定一个包含索引列表的1dnp.ndarray,该索引为True:[1, 2, 4],以及目标np.ndarray:6的长度

如何快速构建实际的np.ndarray,它应该是[False, True, True, False, True, False]

idx = [1,2,3]
s = 6
a = np.zeros(s,dtype=bool)
a[idx] = True

输出:

[False  True  True  True False False]

相关内容

  • 没有找到相关文章