如何计算 np.insert "obj" 参数？

Question

我目前得到以下输入：

1. 数组"x">
2. 将 x 分散到较大数组后应为零的索引列表。

有没有人碰巧有任何快速代码来计算与此对应的"obj"变量。

示例：x = [1,2,3,4]，我想把它变成x = [1, 2, 0, 0, 3, 0, 4]。我的输入是零应该在结果数组中的位置，即

([1,2,3,4], [2, 3, 5]) => [1, 2, 0, 0, 3, 0, 4]

给定数组"x"和索引，其中生成的零应该是将零放在这些索引处的"obj"参数，我如何计算？

Answer 1

创建具有范围差异的新对象，如下所示 -

In [60]: x
Out[60]: array([1, 2, 3, 4])
In [61]: idx
Out[61]: [2, 3, 5]
In [62]: new_obj = idx - np.arange(len(idx))
In [63]: np.insert(x,new_obj,0)
Out[63]: array([1, 2, 0, 0, 3, 0, 4])