For循环中的Matplotlib图例

u0njafvf 于 2023-05-18 发布在其他

关注(0)|答案(2)|浏览(240)

我正在对数据进行分类，并将其绘制在Map上，每个分类都有一个图例，但每次循环时，我都会在图例中得到一条线。我如何才能在图例中为每一个分类获取一行？
注意：我有单独的for循环，以确保较小的圆圈在较大的圆圈上绘图。

sigcorrs = np.random.rand(100,1)

m = Basemap(llcrnrlon=35.,llcrnrlat=30.,urcrnrlon=-160.,urcrnrlat=63.,projection='lcc',resolution='c',lat_1=20.,lat_2=40.,lon_0=90.,lat_0=50.)  
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',zorder=0)
plt.title('Mean Absolute Error')

for a in range(len(clat)):
    if sigcorrs[a] > 0.8:
        X,Y = m(clon[a],clat[a])  
        m.scatter(X,Y,s=300,label='Corr > 0.8')
    else:
        continue

for a in range(len(clat)):
    if sigcorrs[a] > 0.6 and sigcorrs[a] <= 0.8:
        X,Y = m(clon[a],clat[a])  
        m.scatter(X,Y,s=200,label='Corr > 0.6')
    else:
        continue

for a in range(len(clat)):
    if sigcorrs[a] > 0.4 and sigcorrs[a] <= 0.6:
        X,Y = m(clon[a],clat[a])  
        m.scatter(X,Y,s=100,label='Corr > 0.4')
    else:
        continue

for a in range(len(clat)):
    if sigcorrs[a] <= 0.4:
        X,Y = m(clon[a],clat[a])  
        m.scatter(X,Y,s=50,label='Corr < 0.4')
    else:
        continue

plt.legend()
plt.show()

matplotlib

来源：https://stackoverflow.com/questions/34212402/matplotlib-legends-in-for-loop

2条答案

按热度按时间

disho6za1#

您可以通过为每个类别设置一个标签来避免这种情况。例如，在第一个循环中：

label_added =False
for a in range(len(clat)):
    if sigcorrs[a] > 0.8:
        X,Y = m(clon[a],clat[a])  
        if not label_added:
            m.scatter(X,Y,s=300,label='Corr > 0.8')
            label_added = True
        else:
            m.scatter(X,Y,s=300)
    else:
        continue

赞(0）回复(0）举报 2023-05-18

ltqd579y2#

另一种方法是在for循环内部绘制不带标签的数据，并在for循环外部绘制最后一次索引的数据的一个示例。例如：

错误的方式 *

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.colors

example_y1 = np.random.rand(100,1) * 1.2
example_y2 = np.random.rand(100,1)

x = range(len(example_y1))

for i in range(len(example1)):
    if example_y1[i] > example_y2[i]:
        plt.scatter(x[i],example_y1[i,0], c = 'blue', label='example1')
        plt.scatter(x[i],example_y1[i,0], c = 'green', label='example2')

plt.ylabel('Y')
plt.xlabel('X')
plt.legend(loc='upper left')
plt.show()

正确的方式 *

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.colors

example_y1 = np.random.rand(100,1) * 1.2
example_y2 = np.random.rand(100,1)

x = range(len(example_y1))

for i in range(len(example1)):
    if example_y1[i] > example_y2[i]:
        plt.scatter(x[i],example_y1[i,0], c = 'blue')
        plt.scatter(x[i],example_y1[i,0], c = 'green')
        n = i # last index

plt.scatter(x[n],example_y1[n,0], c = 'blue', label='example1')
plt.scatter(x[n],example_y1[n,0], c = 'green', label='example2')
plt.ylabel('Y')
plt.xlabel('X')
plt.legend(loc='upper left')
plt.show()

赞(0）回复(0）举报 2023-05-18

我来回答

For循环中的Matplotlib图例

2条答案

相关问题

热门标签

最新问答