import numpy as np import matplotlib.pyplot as plt def f(t): return np.exp(-t)*np.cos(2*np.pi*t) t1=np.arange(0.0,5.0,0.1) t2=np.arange(0.0,5.0,0.02) plt.figure(1) plt.subplot(211) plt.plot(t1,f(t1),'bo',t2,f(t2),'k') plt.subplot(212) plt.plot(t2,np.cos(2*np.pi*t2),'r--') plt.show()代码解释
时间: 2023-12-02 07:02:36 浏览: 111
import numpy as np import matplotlib.pyplot as plt import math
好的,下面是代码的解释:
```python
import numpy as np
import matplotlib.pyplot as plt
# 定义一个函数 f(t),用于生成数据
def f(t):
return np.exp(-t) * np.cos(2*np.pi*t)
# 生成一组数据 t1,用于在 subplot(211) 中绘制散点图
t1 = np.arange(0.0, 5.0, 0.1)
# 生成一组数据 t2,用于在 subplot(211) 和 subplot(212) 中绘制线条图
t2 = np.arange(0.0, 5.0, 0.02)
# 创建一个新的 figure,用于显示绘图结果
plt.figure(1)
# 创建 subplot(211),用于绘制散点图和线条图
plt.subplot(211)
# 绘制散点图,使用蓝色圆圈表示
plt.plot(t1, f(t1), 'bo')
# 绘制线条图,使用黑色实线表示
plt.plot(t2, f(t2), 'k')
# 创建 subplot(212),用于绘制线条图
plt.subplot(212)
# 绘制线条图,使用红色虚线表示
plt.plot(t2, np.cos(2*np.pi*t2), 'r--')
# 显示绘图结果
plt.show()
```
这段代码的主要功能是绘制两个子图,其中 subplot(211) 中包含散点图和线条图,subplot(212) 中包含线条图。
在 subplot(211) 中,使用 `np.arange()` 函数生成一组数据 t1 和一组数据 t2,分别用于绘制散点图和线条图。然后使用 `plt.plot()` 函数绘制散点图和线条图,其中 `bo` 表示蓝色圆圈,`k` 表示黑色实线。
在 subplot(212) 中,绘制一条以 t2 为横坐标,以 $\cos(2\pi t)$ 为纵坐标的红色虚线。
最后使用 `plt.show()` 函数显示绘图结果。
阅读全文