生成稀疏向量

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. ])