r0 = np.outer([1,1,1],[0,0,0,1,1,1,2,2,2]) bins = np.append(np.append(r0,3+r0, axis=0),6+r0, axis=0).flatten() sums = np.reshape(np.bincount(bins, weights=board.flatten()),[3,3])