python - 如何在python中将二维numpy方形数组旋转45度？

Question

我有一个 numpy 图像数组。维度为 2，形状为 (100,100)。我想增加更多数据，因为我只有 52 组 numpy 数组。我想将给定的数组旋转 45 度。我该怎么办呢？？

假设数组是这样的

a=[[ 0  1  2  3  4]
 [ 5  6  7  8  9]
 [10 11 12 13 14]
 [15 16 17 18 19]
 [20 21 22 23 24]]

请将给定数组旋转 45 度。

Answer 1

您可以使用scipy.ndimage.rotate

import numpy as np
from scipy.ndimage import rotate
x = np.arange(25).reshape(5, -1)
rotate(x, angle=45)

输出

array([[ 0,  0,  0,  0,  0,  0,  0],
       [ 0,  0,  0,  6,  0,  0,  0],
       [ 0,  0,  4,  9, 14,  0,  0],
       [ 0,  3,  8, 12, 16, 21,  0],
       [ 0,  0, 10, 15, 20,  0,  0],
       [ 0,  0,  0, 18,  0,  0,  0],
       [ 0,  0,  0,  0,  0,  0,  0]])