Установить описание легенды как цвет - PullRequest
Новая
       10

Установить описание легенды как цвет

0 голосов
/ 17 апреля 2019

Я строю точечную фигуру в моей таблице.Он имеет значения 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 в виде синей точки на легенде:

Scatter Plot Results

Что мне делать?

1 Ответ

0 голосов
/ 23 апреля 2019

Решено!

Мое плохое в кодировании.Я сделал цикл, чтобы "найти" значения 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()

Вот результат: Рисунок

...