给出（固定）元素的索引时创建一个numpy数组

Question

我有一个numpy索引数组，例如[1,3,12]。我想从这里创建另一个数组，这样在这些索引处，我得到一个非零元素，例如1.所以在这种情况下，使用输入[1,3,12]，我应该得到[0,1,0,1,0,0,0,0,0,0,0,0,1]。我可以在for循环中完成它，是否有一个简短的numpy函数来实现这个目标？

Answer 1

使用numpy，您可以直接使用列表编制索引：

a = [1,3,12]
vector = numpy.zeros(shape=max(a) + 1)
vector[a] = 1