python - 根据标签更改颜色和大小(matplotlib，python 3)

Question

我正在尝试使用 matplotlib 显示散点图，到目前为止显示数据、颜色和大小在硬编码时效果很好，但我需要根据标签集分配大小和颜色，例如，DataSet 看起来像:

data = [[1,1],[1,1],[1,1]]
label =['label1','label3','label3']

散点图应如下所示

关于颜色:

so label1 is green
so label2 is red
so label3 is blue

关于尺寸:每件之间的尺寸差异很小，您可以看出差异。

感谢任何帮助，我正在使用 python 进行编码

谢谢

Answer 1

这是您可以使用的代码片段。这个example from the matplotlib如果您需要更多自定义，文档可能会有所帮助。

import matplotlib.pyplot as plt

data = [[1,2],[2,1],[1,1]]
labels = ['label1','label2','label3']

colorMap = {'label1': 'g',
            'label2': 'r',
            'label3': 'b'}
sizeMap = {'label1': 10,
            'label2': 20,
            'label3': 30}

colors = [ colorMap[label] for label in labels]
sizes = [ sizeMap[label] for label in labels]
x,y = zip(*data)
plt.scatter(x,y, c=colors, s=sizes)
plt.show()