matplotlib绘图气泡图图例
时间: 2024-06-08 18:04:17 浏览: 201
Matplotlib是一个广泛使用的Python数据可视化库,其中创建气泡图(Bubble Plot)可以帮助我们展示三个或四个变量的关系。在气泡图中,每个点代表一个观测值,点的位置由两个数值变量决定,而点的大小(泡泡的大小)和颜色通常用来表示第三个或第四个变量。
为了创建一个带有图例的气泡图,你需要首先安装matplotlib,然后使用`plt.scatter()`函数绘制数据,并用`plt.scatter()`的`c`参数指定颜色编码的变量。对于大小,可以使用`s`参数。至于图例,你可以通过`plt.legend()`添加,并为图例中的不同大小或颜色类别提供标签。
以下是一个简单的例子:
```python
import matplotlib.pyplot as plt
import numpy as np
# 假设我们有三个变量:x, y, size 和 color
x = np.random.rand(100)
y = np.random.rand(100)
size = np.random.rand(100) * 50 # 泡泡的大小范围
color = np.random.rand(100) # 泡泡的颜色
fig, ax = plt.subplots()
ax.scatter(x, y, s=size, c=color, alpha=0.5) # alpha设置透明度
# 添加图例
bubble_legend = {
'size_label': 'Bubble Size', # 大小的标签
'color_label': 'Color Code' # 颜色的标签
}
plt.colorbar(label=bubble_legend['size_label']) # 添加大小的色彩条
legend_elements = [plt.Line2D(, , color=color, lw=0, label=bubble_legend['color_label'])]
ax.legend(handles=legend_elements)
# 显示图例
plt.title("Bubble Plot with Size and Color Legend")
plt.show()
```
阅读全文