將零添加到numpy數組中的給定位置

Question

我有一個numpy數組，並且我想要一個函數以numpy數組和索引列表作為輸入，並返回另一個具有以下屬性的數組作為輸出：將零添加到初始數組，位置為原始數組的每個索引。

讓我舉幾個例子：

如果indices = [1] ，並且初始數組為array([1, 1, 2]) ，則函數的輸出應為array([1, 0, 1, 2]) 。

如果indices = [0, 1, 3]且初始數組為array([1, 2, 3, 4]) ，則函數的輸出應為array([0, 1, 0, 2, 3, 0, 4]) 。

我想以向量化的方式來做，沒有任何for循環。

Answer 1

以前有過同樣的問題。 使用np.insert找到了一個解決方案：

import numpy as np
np.insert([1, 1, 2], [1], 0)

>>> [1, 0, 1, 2]

我看到@jdehesa已經對此發表了評論，但作為將來訪問者的永久答案而添加。