如何重塑存儲彩色圖像序列（RGB）的numpy數組的數據？

Question

我有一個形狀[n_samples，color_channels * time_window，寬度，高度] numpy數組存儲七幀彩色圖像的數據，我想重塑它到[n_samples，color_channels，time_window，width，height]如何可以我這麼做？

Answer 1

第一：形狀是元組。你可以試試，假設A是你原來的陣列：

B = np.reshape(A, (n_samples, color_channels, time_window, width, -1)) 
print(B.shape)