我有一个numpy数组,其维数如下(256、128、4200)-基本上前两个可以形成图像,第三个是通道,第四个是帧(“时间实例”)。如何重塑数组,使帧一个接一个地“堆叠”,换句话说,数组的形状是(256,128*200,4)?重要的是,连接是按帧顺序进行的,因此帧中的值的顺序将保持不变。在
本质上,我们需要优化:
data_new = data[:, :, :, 0]
for i in range(1, data.shape[3]):
data_new = np.concatenate((data_new, data[:, :, :, i]), axis=1)
用^{} 置换轴并重塑-
或者
^{pr2}$roll-axis
并重塑-运行时测试-
因此,
76x+
加速是矢量化利润。在相关问题 更多 >
编程相关推荐