二维线性变换动画python代码
时间: 2023-10-13 12:10:14 浏览: 80
二元线性回归python代码
以下是一个简单的二维线性变换动画的Python代码示例:
```python
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation
# 定义初始点和变换矩阵
x = np.array([1, 1])
A = np.array([[2, -1], [1, 1]])
# 定义动画函数
def animate(i):
global x
x = np.dot(A, x)
plt.cla()
plt.xlim([-5, 5])
plt.ylim([-5, 5])
plt.plot([0, x[0]], [0, x[1]], 'r-', lw=2)
plt.plot(0, 0, 'ko')
plt.plot(x[0], x[1], 'bo')
# 创建动画对象
fig = plt.figure()
ani = animation.FuncAnimation(fig, animate, interval=500)
# 展示动画
plt.show()
```
运行代码后,将会生成一个动画窗口,展示初始点和每次变换后的点。每隔500毫秒,程序将自动执行一次变换,更新点的位置。你可以根据需要调整初始点和变换矩阵,以及动画的帧率和持续时间等参数。
阅读全文