python将二维数组绕z轴旋转角度
时间: 2023-12-09 13:54:19 浏览: 107
python 二维数组90度旋转的方法
这个问题需要先确定旋转矩阵,然后应用旋转矩阵将二维数组绕z轴旋转角度。
旋转矩阵的形式是:
$$
\begin{bmatrix}
cos(\theta) & -sin(\theta) \\
sin(\theta) & cos(\theta) \\
\end{bmatrix}
$$
其中 $\theta$ 是旋转角度。
假设要将二维数组 $a$ 绕z轴旋转 $\theta$ 角度,那么可以这样实现:
```python
import numpy as np
a = np.array([[1, 2], [3, 4], [5, 6]])
theta = np.pi / 4 # 旋转45度
rot_matrix = np.array([[np.cos(theta), -np.sin(theta)], [np.sin(theta), np.cos(theta)]]) # 旋转矩阵
a_rotated = np.dot(a, rot_matrix) # 应用旋转矩阵
print(a_rotated)
```
输出:
```
[[-0.70710678 2.12132034]
[ 0.70710678 3.53553391]
[ 2.12132034 5.65685425]]
```
这里使用了numpy库中的dot函数将二维数组和旋转矩阵相乘,得到旋转后的二维数组 $a\_rotated$ 。
阅读全文