как отобразить надписи легенды на пузырьковой диаграмме matplotlib - PullRequest
Новая
       58

как отобразить надписи легенды на пузырьковой диаграмме matplotlib

0 голосов
/ 31 декабря 2018

Я пытаюсь создать метки легенды на моей пузырьковой диаграмме matplotlib, но не получаю правильные метки

Чтобы построить легенду, я попытался создать цикл для итерации типов городов и соответствующего цвета пузыря.но не получая ожидаемых результатов.

Ожидаемый результат для надписи на легенде должен быть следующим: Городская пузырьковая фигура светло-кораллового цвета Городская пузырьковая фигура светло-синего цвета Пригородная пузырьковая фигура золотого цвета Сельская

ФактическаяРезультаты: Городская фигура пузырькового цвета с светло-коралловым цветом Средний тариф ($) 

%matplotlib inline 
# Dependencies and Setup
import matplotlib.pyplot as plt
import pandas as pd
import numpy as np


# Create a legend
plt.title("Pyber Ride Sharing Data(2016)")
plt.xlabel("Total Number of Rides(Per City)")
plt.ylabel("Average Fare ($)")
plt.grid(True)

unique_bubble_colors = list(xandy_groupby_df[2].unique()) #['LightCoral', 'LightSkyBlue', 'Gold']
 unique_city_types = list(xandy_groupby_df[1].unique())    #['Urban', 'Suburban', 'Rural']

legd = plt.legend(numpoints=1, loc=1, borderpad=0, frameon=True, framealpha=0.9, title="City Types")

 #----------------------------------------------------------------------------
#loop to the unique city types and its corresponding bubble color to print the legend labels - this the part of the code that I am not getting the expected results

i=0
for handle in legd.legendHandles:
    handle.set_sizes([25.0])
    handle.set_color(unique_bubble_colors[i])
    handle.set_label(unique_city_types[i])
    i =+ 1

 #----------------------------------------------------------------------------- 

plt.show()
...