如何提高矩阵块的平均值

时间:2019-04-28 18:08:31

标签: python arrays numpy reshape

我想使这段代码更有效,我是使用numpy array的初学者,因此尽管下面的代码有效,但我觉得有一种比进行当前所有重塑更有效的方法。 这是在馈入矩阵的函数中,无论矩阵的大小如何,我都希望获得2x2块的平均值(但仍可被4整除) M = [[1 2 3 4] [5 6 7 8] [9 10 11 12] [13 14 15 16]]

如果M_new [0]平均为1,2,5,6

M_new = [[3.5 5.5] [11.5 13.5]] 例如:

h_2 = (M.shape[0])//2    
M_new = M.reshape(h_2, 2, -1, 2).swapaxes(1,2).reshape(-1, 2, 2)
M_new = np.floor(np.mean(np.mean(R_new, axis = 1), axis = 1).reshape(h_2,h_2))

0 个答案:

没有答案