Python中np.rot90函数的用法与示例

小蓝 • 2024-11-11 13:43 • 编程

在NumPy中，np.rot90函数是用来对数组进行旋转操作的。具体来说，它可以将数组沿着一个或多个维度旋转90度或180度。

np.rot90函数的语法如下：

numpy.rot90(m, k=1, axes=(0, 1))

其中，参数m表示要旋转的数组；参数k表示旋转的次数，默认为1，即旋转90度，如果值为2则代表旋转180度；参数axes指定旋转的轴，默认为(0, 1)，即在前两个轴上进行旋转。

在使用np.rot90函数进行数组旋转时，我们可以先定义一个二维数组：

import numpy as np
arr = np.arange(9).reshape((3, 3))
print(arr)

输出结果如下：

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

接下来，我们可以使用np.rot90对数组进行旋转：

rotated_arr = np.rot90(arr)
print(rotated_arr)

输出结果如下：

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

我们可以看到，数组按照行的方向进行了旋转90度，同时最后一列变成了第一行。

在使用np.rot90函数时，我们可以通过修改k参数的值来进行多次旋转。例如，我们可以将刚才的数组旋转180度：

rotated_arr = np.rot90(arr, k=2)
print(rotated_arr)

输出结果如下：

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

我们可以看到，数组被旋转了180度，即行和列的方向都被反转了。

在使用np.rot90函数时，我们还可以通过axes参数指定旋转的轴。例如，我们可以将刚才的数组绕着第二个轴进行旋转：

rotated_arr = np.rot90(arr, axes=(1, 0))
print(rotated_arr)

输出结果如下：

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

可以发现，数组也是按照行的方向进行旋转的，但是第二列成为了第一行。

在使用np.rot90函数时，需要注意几个问题：

1. np.rot90函数只能对二维数组进行旋转操作。

2. 旋转操作实际上是沿着第一个和第二个轴进行的。

3. 如果要沿着第三个轴进行旋转，则需要使用axes参数进行指定。

在本文中，我们详细介绍了Python中np.rot90函数的用法和示例。我们讲解了np.rot90函数的语法，以及如何使用它对数组进行旋转操作。此外，我们还介绍了通过k和axes参数来控制旋转的方式，最后，我们就一些注意事项进行了总结。

原创文章，作者：小蓝，如若转载，请注明出处：https://www.506064.com/n/151709.html