我有一个表示空间数据的二维numpy数组。我需要提高其分辨率。我还需要在整个空间内均匀分配值。例如,值
5
会变成:
1.25 1.25
1.25 1.25
我看过imresize但我认为插值选项不适用于此。也许有另一种方式?如果可以的话,我想避免迭代行和列。任何帮助将不胜感激!谢谢!
答案 0 :(得分:3)
简单地除以由其高度和宽度定义的块中的元素数,然后复制/展开。要复制,我们可以使用np.repeat
或np.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]])