Как добавить легенду на график разброса, который имеет назначение цвета

Question

У меня есть список значений x и y и список назначений цветов для каждой точки («зеленый», «синий», «красный» и т. Д.). Все примеры, которые я нашел, создают легенду, основанную на отдельных командах plt.scatter(), которые в дальнейшем достаточно просто plt.legend(). making matplotlib scatter plots from dataframes in Python's pandas. Мой разброс не имеет отдельных рассеиваний для каждой цветной группы. Итак, как мне создать легенду, в которой отображаются цвета каждой группы?

import matplotlib.pyplot as plt 

colors = ["red", "orange", "green", "blue", "purple", "gray"] 
regions = ["Hanoi", "Nha Trang", "Vung Tau", "Phu Quoc", "Quang Ngai", "Saigon"] 
region_colors=dict(zip(regions,colors)) 

grp_color=[] 
for i in data['Region']: 
    grp_color.append(region_colors[i]) 

x_long=data[' Longitude'] 
y_lat=data[" Latitude"] 
plt.scatter(x_long,y_lat,c=grp_color) 
plt.legend(grp_color,regions,loc='right') 

Answer 1

Предполагая, что вы список цветов colors = ["blue", "red", "green"] и список регионов regions = ["Africa", "America", "Australia"] вы можете создать список условных обозначений ручек и использовать его для создания легенды:

handlelist = [plt.plot([], marker="o", ls="", color=color)[0] for color in colors] 
plt.legend(handlelist,regions,loc='right')