python3 矩阵旋转

张千苒 3个月前 38浏览 0评论

Python3 是一门强大的编程语言,也是一个应用广泛的工具。在这门语言中,矩阵旋转是一个十分常见的需求。Python3 中,矩阵旋转可以用一些简单的代码实现。下面我们就来看看具体实现方法。

import numpy as np

def rotate(matrix):
    # 逆时针旋转矩阵
    matrix = np.array(matrix)
    return np.rot90(matrix, k=1)

if __name__ == '__main__':
    matrix = [[1,2,3],[4,5,6],[7,8,9]]
    print(rotate(matrix))

在这段代码中,我们使用了 numpy 库,它是 Python3 中用于科学计算的常用库。我们定义了一个旋转函数 rotate(matrix),它接收一个矩阵作为参数,然后将该矩阵进行逆时针旋转,并返回旋转后的矩阵。

在 rotate 函数中,我们先将矩阵转为 numpy 数组,然后使用 np.rot90() 函数对矩阵进行旋转。该函数中的 k 参数表示旋转的次数,当 k=1 时表示逆时针旋转 90 度。最后,我们将旋转后的矩阵作为函数返回值。

在主函数中,我们定义了一个 3x3 的矩阵 matrix,并将该矩阵作为参数传递给 rotate 函数。然后使用 print() 函数将旋转后的矩阵打印出来。运行这段代码,我们可以得到如下输出:

[[3 6 9]
 [2 5 8]
 [1 4 7]]

这就是我们旋转后的矩阵,它是原始矩阵逆时针旋转 90 度得到的。

总结一下,Python3 中实现矩阵旋转非常简单,只需要使用 numpy 库中的 np.rot90() 函数即可实现。希望这篇文章能够帮助大家快速掌握矩阵旋转的实现方法。