python - 旋转由 numpy 数组处理的图像

Question

我正在研究一个与著名的 MNIST 类似的数据集，除了这个数据集处理阿拉伯字符。我当前拥有的数据采用 numpy 二维数组的形式。看起来像 array([[0, 0, 0, ..., 0, 0, 0] ,..., [0, 0, 0, ..., 0, 0, 0]])形状为(3360, 1024) ，3360 个实例，1024 个功能。

为了显示图像，我将实例的特征 reshape 为 32x32 数组，并使用 matplotlib 绘制 cmap。然而，根据我的数据给出的方式，图像是旋转的，我想将其逆时针旋转 90 度。

有什么建议吗？我想我将不得不操纵每个实例的功能。但是，有没有一种简单的方法可以做到这一点，例如现有的函数。

Answer 1

第 1 步: reshape
第 2 步:旋转

data_rot90 = np.rot90(data.reshape(-1, 32, 32), k=-1, axes=(-2, -1))

这会 reshape 数据，然后沿最后两个轴逆时针旋转 90 度。