在numpy数组中合并轴

Question

我想将 X,Y,Z numpy 数组转换为 (X*Z)*Y numpy 数组。

代码（慢）：

 def rearrange(data):
        samples,channels,t_insts=data.shape
        append_data=np.empty([0,channels])
        for sample in range(0,samples):
            for t_inst in range(0,t_insts):
                channel_data=data[sample,:,t_inst]
                append_data=np.vstack((append_data,channel_data))
        return append_data.shape

如果可能的话，我正在寻找更好的矢量化方法

Answer 1

您可以使用np.transpose 将rows 与columns 交换，然后重新整形-

data.transpose(0,2,1).reshape(-1,data.shape[1])

或者使用np.swapaxes做行列的交换然后reshape -

data.swapaxes(1,2).reshape(-1,data.shape[1])