2017-08-09 196 views
1

嗨我對matplotlib相當新,但我想繪製一個堆積的條形圖。我的酒吧不是堆疊,而是彼此重疊。Matplotlib堆積的條形圖

這是我正在存儲數據的字典。

eventsDict = { 
'A' : [30.427007371788505, 3.821656050955414], 
'B' : [15.308879925288613, 25.477707006369428], 
'C' : [10.846066723627477, 1.910828025477707], 
'D' : [0.32586881793073297, 0.6369426751592357], 
'E' : [3.110656307747332, 11.464968152866243], 
'F' : [8.183480040534901, 1.910828025477707], 
'G' : [3.048065650644783, 16.560509554140125], 
'H' : [9.950920976811652, 4.45859872611465] 
} 

我堆積的條形圖有兩個小節。第一個包含來自列表第一個值的所有數據,第二個包含列表中的所有第二個值。 (名單是在字典中的值)

首先,我的字典轉換爲一個元組:

allEvents = list(self.eventsDict.items()) 

這會把字典到此列表:

all Events = [('A', [30.427007371788505, 3.821656050955414]), ('B', [15.308879925288613, 25.477707006369428]), ('C', [10.846066723627477, 1.910828025477707]), ('D', [0.32586881793073297, 0.6369426751592357]), ('E', [3.110656307747332, 11.464968152866243]), ('F', [8.183480040534901, 1.910828025477707]), ('G', [3.048065650644783, 16.560509554140125]), ('H', [9.950920976811652, 4.45859872611465])] 

這是我繪製它:

range_vals = np.linspace(0, 2, 3) 
    mid_vals = (range_vals[0:-1] + range_vals[1:]) * 0.5 
     colors = ['#DC7633', '#F4D03F', '#52BE80', '#3498DB', '#9B59B6', '#C0392B', '#2471A3', '#566573', '#95A5A6'] 
     x_label = ['All events. %s total events' % (totalEvents), 'Corrected p-value threshold p < %s. %s total events' % (self.pVal, totalAdjusted)] 

    #Turn the dict to a tuple. That way it is ordered and is subscriptable. 
    allEvents = list(self.mod_eventsDict.items()) 
    #print (allEvents) 

    #Use below to index: 
    #list[x] key - value pairing 
    #list[x][0] event name (key) 
    #list[x][1] list of values [val 1(all), val 2(adjusted)] 

    #Plot the Top bar first 
    plt.bar(mid_vals, allEvents[0][1], color = colors[0], label = allEvents[0][0]) 

    #Plot the rest 
    x = 1 
    for x in range(1, 20): 
     try: 
      plt.bar(mid_vals, allEvents[x-1][1], bottom =allEvents[x-1][1], color = colors[x], label = allEvents[x][0])   
      x = x + 1 
     except IndexError: 
      continue 


    plt.xticks(mid_vals) # for classic style 
    plt.xticks(mid_vals, x_label) # for classic style 

    plt.xlabel('values') 
    plt.ylabel('Count/Fraction') 
    plt.title('Stacked Bar chart') 
    plt.legend() 
    plt.axis([0, 2.5, 0, 1]) 
    plt.show() 

這是圖形輸出。理想情況下,堆疊時它們應該全部加起來爲1。我把它們全部做成整體的一小部分,這樣兩根酒吧的高度就會相同。但是,它們只是相互重疊。另外請注意,堆棧與字典上的名稱有不同的標籤。

stacked bar graph output

請幫我調試!

回答

0

您需要以不同的方式設置bottom - 這會告訴matplotlib放置要繪製的小節底部的位置,因此它需要是之前小節的所有高度的總和。

例如,您可以跟蹤條的當前高度與列表,像這樣:

current_heights = [0] * 20 
for x in range(20): 
    try: 
     plt.bar(mid_vals, allEvents[x][1], bottom=current_heights[x], color=colors[x], label=allEvents[x][0])   
     x = x + 1 
     current_heights[x] += allEvents[x][1] #increment bar height after plotting 
    except IndexError: 
     continue 
+0

感謝那定了!我跟蹤了當前的高度,就像你在for循環中增加它一樣。 – Carmelle