生成稀疏向量

Question

我有两份名单。一个带有索引(明显设置)，另一个带有值。是否有可能在固定大小的numpy数组中高效地转换它们？

indexes = [1,2,6,7] 
values = [0.2,0.5,0.6,0.2]
size = 10

我想输出的是：

print(magic_func(indexes,values,size))
array(0,0.2,0.5,0,0,
      0,0.6,0.2,0,0)

Answer 1

如果你想要的话，这很容易分为两行：

In [1]: import numpy as np

In [2]: arr = np.zeros(size)

In [3]: arr[indexes] = values

In [4]: arr
Out[4]: array([ 0. ,  0.2,  0.5,  0. ,  0. ,  0. ,  0.6,  0.2,  0. ,  0. ])