n.median()返回深度为3的数组的1d标量?

时间:2018-09-22 12:50:01

标签: python numpy

我有一个二维的numpy数组。数组中的每个单元格/元素都包含一个表示RGB颜色的标量/元组。当我尝试在数组上计算numpy.median()numpy.average()时,当我应该取回3d标量/元组时,我取回1d标量。

例如;

f = (1,2,3)
a = np.array([[f,f,f],
    [f,f,f]])
print("A: {}, {}".format(np.median(a), a.shape))

输出为:

  

A:2.0,(2、3、3)

中位数不是(1,2,3)而不是2.0吗?

1 个答案:

答案 0 :(得分:1)

您希望np.median(a, (0,1))计算沿前两个轴的中位数。