如果在增加一个标签维度,怎么将气泡上添加标签
时间: 2024-03-09 20:50:41 浏览: 16
如果要在气泡图上添加标签,可以使用Matplotlib库中的annotate函数。annotate函数可以在指定的位置添加一个带有文本和箭头的注释框。以下是一个简单的示例代码:
```python
import matplotlib.pyplot as plt
import numpy as np
# 准备数据
ar_list = [0.8, 0.9, 0.7, 0.6, 0.5]
map_list = [0.7, 0.6, 0.8, 0.9, 0.5]
bubble_size_list = [100, 200, 300, 400, 500]
label_list = ['a', 'b', 'c', 'd', 'e']
# 绘制气泡图
fig, ax = plt.subplots()
scatter = ax.scatter(ar_list, map_list, s=bubble_size_list, alpha=0.5)
# 添加标签
for i, txt in enumerate(label_list):
ax.annotate(txt, (ar_list[i], map_list[i]))
plt.xlabel('AR')
plt.ylabel('mAP')
plt.title('AR vs mAP')
plt.show()
```
运行代码后,应该可以看到一个带有标签的气泡图。annotate函数的第一个参数是要添加的文本,第二个参数是注释框的位置,这里使用了AR和mAP指标的坐标。enumerate函数用于遍历label_list列表,并返回每个元素的索引和值。
注意,如果气泡图上的标签过多,可能会导致标签重叠,影响可读性。在这种情况下,可以考虑使用其他的可视化方式,如热力图。