可以使用Python的NumPy库中的ndarray方法来实现矩阵的旋转。
具体的实现步骤如下:
import numpy as np
def rotate_matrix(X, degree, direction):
if direction == 1:
# 逆时针旋转
X_rotated = np.rot90(X, k=int(degree/90))
else:
# 顺时针旋转
X_rotated = np.rot90(X, k=int(4-degree/90))
return X_rotated
完整代码如下:
import numpy as np
def rotate_matrix(X, degree, direction):
if direction == 1:
# 逆时针旋转
X_rotated = np.rot90(X, k=int(degree/90))
else:
# 顺时针旋转
X_rotated = np.rot90(X, k=int(4-degree/90))
return X_rotated
示例代码:
# 定义一个3x3的矩阵
X = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 将矩阵逆时针旋转90度
X_rotated_1 = rotate_matrix(X, 90, 1)
print(X_rotated_1)
# 将矩阵顺时针旋转180度
X_rotated_2 = rotate_matrix(X, 180, -1)
print(X_rotated_2)
输出:
[[3 6 9]
[2 5 8]
[1 4 7]]
[[9 8 7]
[6 5 4]
[3 2 1]]