給定（固定）元素的索引時創建一個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