how to slice and extend a 2d numpy array?,"q.T.reshape(-1, k, n).swapaxes(1, 2).reshape(-1, k)"