使用matplotlib库绘制极坐标曲线图

作者:梅琳marlin2024.01.17 22:23浏览量:10

简介:介绍如何使用matplotlib库的add_subplot()函数,绘制极坐标曲线图。

在使用matplotlib库绘制极坐标曲线图时,可以使用add_subplot()函数来创建一个子图,并将其设置为极坐标投影。以下是一个简单的示例代码,演示如何使用add_subplot()函数绘制极坐标曲线图。
首先,我们需要导入matplotlib库和numpy库。

  1. import matplotlib.pyplot as plt
  2. import numpy as np

接下来,我们可以创建一个数据集,例如一组角度和对应的半径值。

  1. theta = np.linspace(0, 2*np.pi, 100) # 角度值
  2. r = np.sin(theta) # 半径值

然后,我们可以使用add_subplot()函数创建一个子图,并将其设置为极坐标投影。

  1. fig = plt.figure() # 创建画布
  2. ax = fig.add_subplot(111, projection='polar') # 创建子图并设置为极坐标投影

接下来,我们可以使用plot()函数绘制极坐标曲线图。

  1. ax.plot(theta, r) # 绘制极坐标曲线图

最后,我们可以显示绘制的图形。

  1. plt.show() # 显示图形

完整的代码如下所示:
```python
import matplotlib.pyplot as plt
import numpy as np
theta = np.linspace(0, 2*np.pi, 100) # 角度值
r = np.sin(theta) # 半径值
fig = plt.figure() # 创建画布
ax = fig.add_subplot(111, projection=’polar’) # 创建子图并设置为极坐标投影
ax.plot(theta, r) # 绘制极坐标曲线图
plt.show() # 显示图形