中位数组合适合python中的图像

时间:2012-12-06 21:50:44

标签: python multidimensional-array fits

我有二维numpy数组形式的三个拟合图像。我想中间组合它们,即生成一个输出数组,其中每个像素是三个输入数组中相同像素的中值。这可以使用imcombine在IRAF上轻松完成。有没有办法在Python上执行此操作而不循环遍历整个数组并获取每个像素的中位数?

1 个答案:

答案 0 :(得分:12)

最简单的方法是:

  • 堆叠2d数组以形成3d数组
  • 使用numpy.median传递axis=0来计算中位数,以计算堆叠维度。

你实质上是计算元素中位数。这是我要做的一个简单的例子:

>>> import numpy
>>> a = numpy.array([[1,2,3],[4,5,6]])
>>> b = numpy.array([[3,4,5],[6,7,8]])
>>> c = numpy.array([[9,10,11],[12,1,2]])
>>> d = numpy.array([a,b,c])
>>> d
array([[[ 1,  2,  3],
        [ 4,  5,  6]],

       [[ 3,  4,  5],
        [ 6,  7,  8]],

       [[ 9, 10, 11],
        [12,  1,  2]]])
>>> d.shape
(3, 2, 3)

>>> numpy.median(d, axis=0)
array([[ 3.,  4.,  5.],
       [ 6.,  5.,  6.]])