根據標簽更改顏色和大小（matplolib，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

這是您可以使用的代碼段。 如果需要更多自定義，則來自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()