如何在 python 中使用 matplotlib 为图中的每个散点获取不同的标签名称？

Question

我正在使用 for 循环为点提供标签和颜色。 但是我没有得到每个不同颜色符号的唯一标签，相反，由于 500 行左右的大型 Excel 数据，我在图表顶部排起了长队。 这是我的代码：

for val in ghi:
if val < 2:
    col.append('navy')
    lab.append(' <2 ')
elif (val >= 2) and (val < 4):
    col.append('lightblue')
    lab.append(' 2~4 ')
elif (val >= 4) and (val < 6):
    col.append('orange')
    lab.append(' 4~6 ')
else:
    col.append('brown')
    lab.append(' >6 ')
plt.scatter(x, y, marker = "D", s = 10, color = col, label = lab)
sns.lineplot(x, y, color ="red")
plt.legend()

Answer 1

您看到多个标签的原因是label=lab 。 每行有一个条目。 因此，您将需要使用plt.legend()添加句柄和标签，这样您就可以像 if/else 语句中那样只有四个条目。 此外，为确保颜色和标签匹配，请按实验室字段对数据进行排序。

由于没有提供任何数据，前几行包括随机数据的创建。 请注意，我将数据放入数据框中并按ghi字段对其进行排序。 稍后，图例具有您可以根据需要自定义的句柄和标签。 希望这是您想要实现的目标。

否则，我已尝试保持相同的名称和尽可能多的代码，以便您更容易解释。

import seaborn as sns
import matplotlib.pyplot as plt
import pandas as pd
import numpy as np
from matplotlib.lines import Line2D 

ghi = np.random.randint(11, size=(20))
x = np.random.rand(20)
y = np.random.rand(20)
col = []
for val in ghi:
    if val < 2:
        col.append('navy')
    elif (val >= 2) and (val < 4):
        col.append('lightblue')
    elif (val >= 4) and (val < 6):
        col.append('orange')
    else:
        col.append('brown')

df = pd.DataFrame({'x':x, 'y':y, 'ghi':ghi, 'col':col})
df.sort_values(by=['ghi'], inplace = True)
plt.scatter(df.x, df.y, marker = "D", s = 50, color = df.col)#, label = np.unique(lab))
sns.lineplot(x, y, color ="red")
myHandles = [Line2D([], [], marker='D', color='navy', linestyle='None'),
          Line2D([], [], marker='D', color='lightblue', linestyle='None'),
          Line2D([], [], marker='D', color='orange', linestyle='None'),
          Line2D([], [], marker='D', color='brown', linestyle='None')]

plt.legend(handles=myHandles, labels = [' <2 ', ' 2~4 ', ' 4~6 ', ' >6 '])

输出图