增加一个numpy数组并均匀分配值

时间:2017-09-14 09:29:30

标签: python arrays numpy

我有一个表示空间数据的二维numpy数组。我需要提高其分辨率。我还需要在整个空间内均匀分配值。例如,值

5

会变成:

1.25 1.25

1.25 1.25

我看过imresize但我认为插值选项不适用于此。也许有另一种方式?如果可以的话,我想避免迭代行和列。任何帮助将不胜感激!谢谢!

1 个答案:

答案 0 :(得分:3)

简单地除以由其高度和宽度定义的块中的元素数,然后复制/展开。要复制,我们可以使用np.repeatnp.lib.stride_tricks.as_strided

使用np.repeat -

def upscale_repeat(a, h, w):
    return (a/float(h*w)).repeat(h, axis=0).repeat(h, axis=1)

np.lib.stride_tricks.as_strided使用tile_array -

def upscale_strided(a, h, w):
    return tile_array(a/float(h*w), h, w)

示例运行 -

In [140]: a
Out[140]: 
array([[ 7,  6,  9],
       [ 6,  6, 10]])

In [141]: upscale_repeat(a, 2, 2)
Out[141]: 
array([[ 1.75,  1.75,  1.5 ,  1.5 ,  2.25,  2.25],
       [ 1.75,  1.75,  1.5 ,  1.5 ,  2.25,  2.25],
       [ 1.5 ,  1.5 ,  1.5 ,  1.5 ,  2.5 ,  2.5 ],
       [ 1.5 ,  1.5 ,  1.5 ,  1.5 ,  2.5 ,  2.5 ]])

In [142]: upscale_repeat(a, 2, 3)
Out[142]: 
array([[ 1.17,  1.17,  1.17,  1.  ,  1.  ,  1.  ,  1.5 ,  1.5 ,  1.5 ],
       [ 1.17,  1.17,  1.17,  1.  ,  1.  ,  1.  ,  1.5 ,  1.5 ,  1.5 ],
       [ 1.  ,  1.  ,  1.  ,  1.  ,  1.  ,  1.  ,  1.67,  1.67,  1.67],
       [ 1.  ,  1.  ,  1.  ,  1.  ,  1.  ,  1.  ,  1.67,  1.67,  1.67]])