本文主要是介绍数据集的图例怎么生成,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
数据集有13个分类,每个分类都有颜色,怎么生成色块的图例。
import matplotlib.pyplot as plt
import matplotlib.patches as mpatchess3dis_color = [[255, 248, 220], [220, 220, 220], [139, 71, 38], [238, 197, 145], [70, 130, 180], [179, 238, 58], [110, 139, 61], [105, 105, 105], [0, 0, 128], [205, 92, 92], [244, 164, 96], [147, 112, 219], [255, 228, 225]]labels_dict = {"0_ceiling": 0,"1_floor": 1,"2_wall": 2,"3_beam": 3,"4_column": 4,"5_window": 5,"6_door": 6,"7_table": 7,"8_chair": 8,"9_sofa": 9,"10_bookcase": 10,"11_board": 11,"12_clutter": 12
}labels = labels_dict.keys()# 创建一个图例对象
legend_elements = []
for color, label in zip(s3dis_color, labels):# 将RGB颜色值转换为范围在[0,1]之间的浮点数color = [c / 255 for c in color]# 创建填充颜色的矩形rect = mpatches.Rectangle((0, 0), 1, 1, facecolor=color)# 创建图例元素并添加到图例对象中legend_elements.append((rect, label))# 创建图例
fig, ax = plt.subplots()legend = ax.legend(*zip(*legend_elements), loc='center', ncol=3)# 去除坐标轴
ax.axis('off')# 保存图例
plt.savefig("legend.png", bbox_inches='tight', pad_inches=0)
这是S3DIS数据集的图例。
![](https://img-blog.csdnimg.cn/direct/58303c457bba44538be83c2424992234.png#pic_center)
这篇关于数据集的图例怎么生成的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!