Matplotlib底图中的传说

Question

我正在底图中绘制多个点，并希望有一个图例来表示每种颜色所代表的类别。但是，由于每个类别中都有多个点，因此图例会拉出每个点，在图例中为我提供相同类别的多个条目。有没有办法只显示一个整体颜色类别列表？

m = Basemap(llcrnrlon=30.,llcrnrlat=20.,urcrnrlon=-160.,urcrnrlat=63.,projection='lcc',resolution='c',lat_1=20.,lat_2=40.,lon_0=90.,lat_0=50.)

X,Y = m(lon,lat)    
m.drawcountries()
m.drawmapboundary(fill_color='lightblue')
m.drawparallels(np.arange(0.,90.,5.),color='gray',dashes=[1,3],labels=[1,0,0,0])
m.drawmeridians(np.arange(0.,360.,15.),color='gray',dashes=[1,3],labels=[0,0,0,1])
m.fillcontinents(color='beige',lake_color='lightblue')
plt.title('MERRA-Observation Correlations')

for j in range(len(corr)):
    if j == 0 or j == 1 or j ==2:
        m.plot(X[j],Y[j],'o',color='orange',markersize=np.absolute(corr[j])*17.5,label='Prairie')
    if j == 3 or j ==4 or j == 5:
        m.plot(X[j],Y[j],'o',color='violet',markersize=np.absolute(corr[j])*17.5,label='Tundra')
    if j ==6 or j == 7 or j == 8:
        m.plot(X[j],Y[j],'o',color='purple',markersize=np.absolute(corr[j])*17.5,label='Taiga')
    plt.legend()

注意：我在循环内外都放置了plt.legend，结果相同。

Answer 1

您似乎正在为每个数据点绘制一个全新的行。散点图可能更合适：尝试：

而不是j循环

scale = 17.5
m.scatter(X[:3], Y[:3], color='orange', s=abs(corr[:3])*scale, label='Prairie')
m.scatter(X[3:6], Y[3:6], color='violet', s=abs(corr[3:6])*scale, label='Tundra')
m.scatter(X[6:], Y[6:], color='purple', s=abs(corr[6:])*scale, label='Taiga')

（未经测试：我没有您的数据）。

Answer 2

您可以保留m.plot返回的对象的句柄，而不是传递图例kwarg。然后，您可以仅使用要保留的图表手动创建图例。

how to add legend for scatter()?有一个很好的例子。

Answer 3

另一种方法：将此呼叫替换为plt.legend()并在外部 for for循环执行：

ax = plt.gca()
handles, labels = ax.get_legend_handles_labels()
legend = plt.legend([handles[0],handles[3],handles[6]], labels[0],labels[3],labels[6]])

这将由plot（scatter等）调用设置的内容分别发送给legend，因此您可以看到如何以不同方式更改图例。

xnx的分散解决方案看起来也不错。

P.S。 - 这与底图没有任何关系，它通常用于matplotlib绘图。