如何计算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])