温馨提示×

python旋转矩阵怎么实现

小亿
110
2023-10-21 11:38:11
栏目: 编程语言

要实现矩阵的旋转,可以使用numpy库中的rot90函数。下面是一个简单的示例代码:

import numpy as np

# 创建一个2x3的矩阵
matrix = np.array([[1, 2, 3],
                   [4, 5, 6]])

# 将矩阵逆时针旋转90度
rotated_matrix = np.rot90(matrix, k=1)

print(rotated_matrix)

输出结果为:

[[3 6]
 [2 5]
 [1 4]]

其中,k参数控制旋转的次数,1表示逆时针旋转90度,2表示逆时针旋转180度,以此类推。

如果你不想使用外部库,可以手动编写代码来实现矩阵的旋转。以下是一个示例代码:

def rotate_matrix(matrix):
    rotated_matrix = []
    for i in range(len(matrix[0])):
        row = [matrix[j][i] for j in range(len(matrix)-1, -1, -1)]
        rotated_matrix.append(row)
    return rotated_matrix

# 创建一个2x3的矩阵
matrix = [[1, 2, 3],
          [4, 5, 6]]

# 将矩阵逆时针旋转90度
rotated_matrix = rotate_matrix(matrix)

print(rotated_matrix)

输出结果为:

[[3, 6],
 [2, 5],
 [1, 4]]

上述代码通过遍历原矩阵的列来生成旋转后的矩阵的行。每一行都是原矩阵对应列的倒序。

0