我正在尝试为Python找到一个等效的代码来设置我在R中设置的以下维度：

dim(train_array) = c(28, 28, 1, ncol(train_x))

train_x 是一个矩阵，包含28×28的缩放后的（灰度）Olivetti人脸图像，这些图像被数字化为784（28 * 28）像素行（每行标记为pixel.1, pixel.2，等等），共有15列。有没有Python包可以像R中的dim()函数那样使用？

回答：

你需要的是pandas的shape函数。只需使用

train_array.shape

如果你想获取数据框的形状，或者

train_array.reshape(28, 28, 1, len(train_x))

如果你想改变数据框的形状。