2011-03-23 40 views
1

有谁知道为什么这段代码不能正确地绘制框的颜色。我希望每个组件都是不同的颜色,但它们都以黑色的图例出现。yplot颜色不是预期的

from numpy import array, zeros 
import matplotlib.pyplot as plt 

# Components: Useage times (start, stop), wattage, detail 

COMPONENTS = {"fridge": ([(0.0,24.0)], 35, " Litres"), 
       "kettle": ([(7.3,7.33), (19.0,19.3)], 2500, ""), 
       "netbook": ([(8.0,9.3),(12.0,15.0)], 12.5, ""), 
       "light bulb": ([(18.0,22.0)], 20, "") 
       } 
COLORS = ('b','g','r','c','m','y','k','w') 
PLOT = [] 
TIME = range(24*60) 

Powers = [] # list of array of power for each component 
for key in COMPONENTS.keys(): # each useage 
    p = zeros(len(TIME)) 
    for j in COMPONENTS[key][0]: # start and stop 
     start = round(j[0]*60) 
     end = round(j[1]*60) 
     p[start:end] = COMPONENTS[key][1] 
    Powers.append(p) 

b=zeros(len(TIME)) 
for i in range(len(COMPONENTS.keys())): 
    PLOT.append(plt.bar(TIME,Powers[i],width = 1, color=COLORS[i], bottom=b)) 
    b+=Powers[i] 


plt.ylabel('Power (W)') 
plt.xlabel('Time (h)') ### 
plt.title('Power Cycle') 
plt.xticks(range(0,25*60,60) ,[str(t) for t in range(25)]) 
plt.legend(tuple([i for i in PLOT]), tuple([c for c in COMPONENTS.keys()])) 

plt.show() 
+1

是什么给了你使用全部大写变量名的想法?这伤害了我的眼睛! – 2011-03-23 15:03:49

回答

2

就图表而言,您的代码很好。问题是有太多的酒吧,所以你只能看到黑色的边框。这是在变焦时什么样子: enter image description here

为了得到传说中的权利,通过label参数而绘制,然后只需调用plt.legend()不带参数时,即可大功告成。

+0

不错,谢谢 – wookie1 2011-03-23 15:20:17