熊猫matplotlib.pyplot添加图例列值

我正在使用熊猫做一个散点图。我的数据是这样的：熊猫matplotlib.pyplot添加图例列值

Locations Lenovo Global Region Primary Function Subsidiaries Apr_2015_to_Mar_2016_[kWh] Apr_2015_to_Mar_2016_[MT] color MT/kWh

263 United Kingdom - Hook EMEA Large Office (OFL) Lenovo 7.561727e+04 129.438515 r 0.001712 
202 South Africa - Johannesburg/Bryanston EMEA Small Office (OSL) Lenovo 1.013746e+05 93.872885 r 0.000926 
232 India - Chennai Factory Asia Pacific Manufacturing (MFG) Motorola Mobility 3.163600e+05 271.933953 g 0.000860 
159 India - Pondicherry Asia Pacific Manufacturing (MFG) Lenovo 1.074016e+06 907.869649 g 0.000845 
242 Australia - Chatswood Asia Pacific Large Office (OFL) Lenovo 3.001254e+05 239.500093 g 0.000798

（对不起，我尽我所能去格式化的表格）。

定义为不同地区使用不同颜色的功能。

def colorpoint(row): 
    if row['Lenovo Global Region'] == 'Asia Pacific': 
     return('g') 
    if row['Lenovo Global Region'] == 'EMEA': 
     return('r') 
    else: 
     return('b') 
test3['color'] = test3.apply (lambda row: colorpoint (row),axis=1)

定义了我想绘制的散点。

y=test3['Apr_2015_to_Mar_2016_[MT]'] 
x=test3['Apr_2015_to_Mar_2016_[kWh]'] 
T = test3['color'] 
area= (y/x)*500000 
xmax=1.1*max(test3['Apr_2015_to_Mar_2016_[kWh]']) 
ymax=1.1*max(test3['Apr_2015_to_Mar_2016_[MT]'])

绘制该图。 FIG1 = plt.figure（figsize =（16,9），DPI = 300） AX = plt.subplot（111）

plot=plt.scatter(x,y,alpha=0.6,c=T,s=area) 
ax.grid(True) 
ax.set_xlim([0,xmax]) 
ax.set_ylim([0,ymax]) 
ax.set_xlabel('Apr 2015 to Mar 2016 [kWh]') 
ax.set_ylabel('Apr 2015 to Mar 2016 [MT]') 
ax.set_title('Total Elec consumtion [kWh] VS CO2 emission [MT]')

尝试添加图例。 我想显示颜色对应于他们的“联想全球地区”，但它不工作，只显示一个区域“美国集团”为蓝点

legend=test3['Lenovo Global Region'] 
plt.legend(legend,loc=4)

感谢，如果你有想法！

来源

2016-07-06 Yang

也许这就是你需要在最后一行的内容：

plt.legend(legend.values,loc=4)

来源

2016-07-06 22:59:14

它不工作，还绘制了同样的事情。 – Yang

熊猫matplotlib.pyplot添加图例列值

回答

相关问题