python - 将较小的数组放入大数组中，同时忽略零

Question

我有一个大数组，想要在某个偏移处放置一个较小的数组，同时忽略较小数组中的零。最好的方法是什么？

我尝试了屏蔽数组，但不知怎的，这很慢

mask = np.ma.masked_equal(pixels, 0, False)
output[offset_y:offset_y+tile_height,offset_x:offset_x+tile_width] = np.where(mask.mask, output[offset_y:offset_y+tile_height,offset_x:offset_x+tile_width], pixels)

然后我尝试了这个

np.place(output[offset_y:offset_y+tile_height,offset_x:offset_x+tile_width], pixels>0, pixels[pixels>0])

但是速度很慢

最快的方法是什么？

Answer 1

我怀疑常规的旧逻辑索引在这里是最有效的:

# make a view into output with the same dimensions as `pixels`
output_subarray = output[offset_y:offset_y+tile_height,offset_x:offset_x+tile_width]

mask = pixels != 0
output_subarray[mask] = pixels[mask]