### Python 中实现数组旋转特定角度的方法 在 Python 中，可以使用 `numpy` 库来处理数组的旋转操作。如果需要实现数组旋转特定角度（例如 90°、180° 或其他任意角度），可以通过以下方法完成： #### 方法一：使用 `numpy` 的内置函数 对于简单的 90°、180° 或 270° 旋转，可以直接使用 `numpy.rot90` 函数[^1]。该函数允许指定旋转次数和轴方向，从而实现矩阵的旋转。 ```python import numpy as np # 示例矩阵 A = np.asfarray([[1, 2], [3, 4]]) # 顺时针旋转 90° B = np.rot90(A, k=-1) # k=-1 表示顺时针旋转 90° # 逆时针旋转 90° C = np.rot90(A) # 默认 k=1 表示逆时针旋转 90° # 旋转 180° D = np.rot90(A, k=2) # k=2 表示旋转 180° ``` #### 方法二：使用图像处理库 `OpenCV` 如果需要实现任意角度的旋转（如 45° 或其他非整数倍角度），可以借助 `OpenCV` 库中的 `cv2.getRotationMatrix2D` 和 `cv2.warpAffine` 函数[^2]。这种方法适用于二维数组或图像数据的旋转。 ```python import cv2 import numpy as np # 示例矩阵 A = np.array([[1, 2], [3, 4]], dtype=np.uint8) # 获取矩阵的中心点 (h, w) = A.shape[:2] center = (w // 2, h // 2) # 定义旋转角度和缩放比例 angle = 45 # 旋转角度 scale = 1.0 # 缩放比例 # 获取旋转矩阵 M = cv2.getRotationMatrix2D(center, angle, scale) # 执行旋转操作 rotated = cv2.warpAffine(A, M, (w, h)) ``` #### 方法三：自定义旋转算法 如果不想依赖外部库，也可以通过数学计算实现数组的旋转。具体步骤包括： 1. 计算旋转矩阵。 2. 使用插值法处理旋转后的像素位置。 以下是一个简单的实现示例： ```python import math import numpy as np def rotate_array(arr, angle): # 获取数组尺寸 h, w = arr.shape[:2] center_x, center_y = w / 2, h / 2 # 创建旋转矩阵 radian = math.radians(angle) cos_val = math.cos(radian) sin_val = math.sin(radian) # 初始化结果数组 result = np.zeros_like(arr) for y in range(h): for x in range(w): # 计算原始坐标相对于中心点的位置 dx = x - center_x dy = y - center_y # 应用旋转矩阵 new_x = int(dx * cos_val + dy * sin_val + center_x) new_y = int(-dx * sin_val + dy * cos_val + center_y) # 检查新坐标是否在范围内 if 0 <= new_x < w and 0 <= new_y < h: result[new_y, new_x] = arr[y, x] return result # 示例矩阵 A = np.array([[1, 2], [3, 4]], dtype=np.uint8) # 旋转 45° rotated = rotate_array(A, 45) ``` ### 注意事项 - 如果输入数组不是 `numpy` 类型，需要先将其转换为 `numpy` 数组，例如使用 `np.asfarray` 或 `np.asarray`[^1]。 - 对于任意角度的旋转，建议使用 `OpenCV`，因为它提供了高效的图像处理功能，并能保持输出图像的大小不变[^2]。