Question

我正在尝试将值填充到numpy数组中。该数组最初用1填充，我的目标是用另一个数组中的值覆盖指定索引处的数值。

import numpy as np
# get initial array of ones
mask = np.ones(10)
# get values to overwrite ones at indices
values = [10, 30, 50.5]
# get indices for which values will replace ones 
idx_pad = [1, 6, 7]

print(mask)
>> [ 1.  1.  1.  1.  1.  1.  1.  1.  1.  1.]

我想得到的是：

>> [ 1  10  1  1  1  1  30  50.5  1  1 ]

我认为使用OrderedDict可以做到这一点，尽管我仍在试图解决这个问题。我也希望通过numpy有一个快速的方法。我希望将此示例应用于我的实际数据集，len(idx_pad) = 10322和len(mask) = 69268。任何帮助将不胜感激。

Answer 1

这是@Divakar的解决方案。

import numpy as np
# get initial array of ones
mask = np.ones(10)
# get values to overwrite ones at indices
values = [10, 30, 50.5]
# get indices for which values will replace ones
idx_pad = [1, 6, 7]

print(mask)
>> [ 1.  1.  1.  1.  1.  1.  1.  1.  1.  1.]

# replace values at indices in idx_pad
mask[idx_pad] = values

print(mask)
>> [  1.   10.    1.    1.    1.    1.   30.   50.5   1.    1. ]

如何在数组1中使用数组2中的值将值填充到数组1中？

1 个答案: