将图例描述设置为颜色表示

Posted 2023-02-16

【中文标题】将图例描述设置为颜色表示

【英文标题】：Set legend description as color representated

【发布时间】：2019-09-07 22:36:37

【问题描述】：

我正在绘制我的数据表的散点图。它具有一段时间内不同建筑物产生的KW值。

散点图的每种颜色都代表一座建筑物。

我不知道该怎么做才能用建筑物和图形上的颜色绘制图例。

这是我的绘图代码：

def plot_data(dataframe, x_values, y_values):
    plt.scatter(dataframe[x_values], dataframe[y_values], alpha=0.5, c=dataframe['ID_SEDE'])
    plt.title('Scatter plot ' + y_values)
    plt.xlabel(x_values)
    plt.ylabel(y_values)
    plt.legend(T_MEDIDAS_CO_DROP_FILTERED['ID_SEDE'].unique())
    plt.show()

该图仅将我的第一个 ID_SEDE 显示为图例上的蓝点：

我该怎么办？

【参考方案1】：

解决了！

我对编码不好。我做了一个循环，以便“定位”每个建筑物的 df 值并将其添加到散点图：

buildings = dataframe['ID_SEDE'].unique()
plt.figure(figsize=(15, 15), dpi=80)
for building in buildings:
    dataframe_build = dataframe.loc[(dataframe['ID_SEDE'] == building)]
    plt.scatter(dataframe_build[x_values], dataframe_build[y_values], alpha=0.5, label=('SEDE ' + str(sede)))
    plt.title('Scatter plot ' + y_values)
    plt.xlabel(x_values)
    plt.xlabel(y_values)
    plt.legend()
plt.show()

这是输出： Figure