我对Python还很陌生,目前正在处理一个代码,在这个代码中,我想要存储一个三维矩阵的先前迭代版本,每次for循环都会创建一个新的版本。我想要采取的方法是通过连接一个新的3+1=4维数组来存储先前的值。现在,这是可行的,并且我已经让它像这样工作:
import numpy as np
matrix = np.ones((1,lay,row,col), dtype=np.float32)
for n in range(100):
if n == 0:
# initialize the storage matrix
matrix_stored = matrix
else:
# append further matrices in first dimension
matrix_stored = np.concatenate((matrix_stored,matrix),axis = 0)
这是我的问题:上面的代码要求矩阵已经是四维结构[1 x m x n x o]。然而,为了我的目的,我希望将变量矩阵保留为三维[m x n x o],并且只有在将其输入到变量matrix_stored时才将其转换为四维形式。
是否有一种方法可以简化这样的转换?