легенда пиплета для графика рассеяния, окрашенная значениями

Question

Когда я использую переменную для раскрашивания графика рассеяния, как я могу сделать легенду о том, какие цвета представляют? Как я могу сделать легенду показать ярлык 0 представляет пустой и 1 представляет полный?

import matplotlib.pyplot as plt 
X = [1,2,3,1,2,3,4] 
Y = [1,1,1,2,2,2,2] 
label = [0,1,1,0,0,1,1] 
plt.scatter(X, Y, c= label, s=50) 
plt.show() 

Answer 1

Дайте этот код попробовать:

import matplotlib.pyplot as plt 
import matplotlib.patches as mpatches 
X = [1, 2 ,3, 1, 2, 3, 4] 
Y = [1, 1, 1, 2, 2, 2, 2] 
labels = [0, 1, 1, 0, 0, 1, 1] 
key = {0: ('red', 'empty'), 1: ('green', 'full')} 
plt.scatter(X, Y, c=[key[index][0] for index in labels], s=50) 
patches = [mpatches.Patch(color=color, label=label) for color, label in key.values()] 
plt.legend(handles=patches, labels=[label for _, label in key.values()], bbox_to_anchor=(1, .3)) 
plt.show() 

И это то, что вы получите:

Чтобы использовать цвета или этикетки отличаются от показанных на рисунке вам просто нужно соответствующим образом изменить значения словаря key.