matplotlib.pyplot as plt 雷达图
时间: 2023-12-21 08:32:08 浏览: 89
python使用matplotlib绘制雷达图
以下是使用matplotlib.pyplot绘制雷达图的示例代码:
```python
import numpy as np
import matplotlib.pyplot as plt
# 创建数据
categories = ['A', 'B', 'C', 'D', 'E']
values = [4, 3, 5, 2, 1]
# 计算角度
angles = np.linspace(0, 2 * np.pi, len(categories), endpoint=False).tolist()
angles += angles[:1]
# 绘制雷达图
fig, ax = plt.subplots(figsize=(6, 6), subplot_kw=dict(polar=True))
ax.fill(angles, values, color='skyblue', alpha=0.5)
ax.set_xticks(angles[:-1])
ax.set_xticklabels(categories)
ax.set_yticks([1, 2, 3, 4, 5])
ax.set_ylim(0, 5)
# 添加标题和网格线
plt.title("Radar Chart")
plt.grid(True)
# 显示图形
plt.show()
```
这段代码首先导入了numpy和matplotlib.pyplot库。然后,创建了一个包含5个类别和对应值的数据。接下来,计算了每个类别对应的角度,并将最后一个角度与第一个角度相同,以闭合雷达图。然后,创建了一个极坐标子图,并使用fill函数绘制了雷达图。最后,设置了刻度、标题和网格线,并显示了图形。
阅读全文