Interaktives gestapeltes Balkendiagramm mit MatplotLib python

Question

Ich brauche, um ein gestapeltes Balkendiagramm mit Matplotlib zu erstellen. Jeder Balken sollte ein Stapel der Parameter sein, die ich messe. Allerdings möchte ich, dass es interaktiv oder dynamisch ist, so dass, wenn ich auf einen der Parameter (A, B, C) zum Beispiel in der Legende klicken, sollte es machen, dass Parameter am unteren Rand des Stapels, so wie wir einen besseren Vergleich zwischen verschiedenen Kandidaten je nach dem Parameter haben wir wählen.

Ich habe von Beispielen in matplotlib inspiriert.. hier ist mein Code

import numpy as np
import matplotlib.pyplot as plt

N = 10  #could change

plt.figure()

A   = np.array([70, 88, 78, 93, 99, 58, 89, 66, 77, 78])
B = np.array([73, 65, 78, 87, 97, 57, 77, 88, 69, 78])
C = np.array([66, 98, 88, 67, 99, 88, 62, 70, 90, 73])

ind = np.arange(N)    # the x locations for the groups
width = 0.35       # the width of the bars: can also be len(x) sequence

p1 = plt.bar(ind, A,width, color='r')
p2 = plt.bar(ind, B, width, color='y', bottom=A)
p3 = plt.bar(ind, C, width, color='b', bottom=A+B)

plt.ylabel('Scores')
plt.title('Index')

plt.xticks(ind+width/2., ('1', '2', '3', '4', '5', '6', '7', '8', '9', '10'))#dynamic - fed

plt.yticks(np.arange(0,300,10))
plt.legend( (p1[0], p2[0], p3[0]), ('A','B','C') )
plt.grid(True)

plt.show()

Ich danke Ihnen... Ich hoffe, ich bin klar genug

Answer 1

7 Jahre später, aber ich dachte, ich würde eine Lösung für dieses Problem mit Python 3.6.0 und Matplotlib 2.0.0 anbieten

import numpy as np
import matplotlib.pyplot as plt

#Variables
bar_a = np.array([70, 88, 78, 93, 99, 58, 89, 66, 77, 78])
bar_b = np.array([73, 65, 78, 87, 97, 57, 77, 88, 69, 78])
bar_c = np.array([66, 98, 88, 67, 99, 88, 62, 70, 90, 73])
xloc = np.arange(len(bar_a)) #Get a count of postions in bar_a instead of declaring 10.
width = 0.5  # the width of the bars: can also be len(x) sequence

#Plotting
fig = plt.figure(figsize=(6,6))
p1 = plt.bar(xloc, bar_a, width, color='r', label='A')
p2 = plt.bar(xloc, bar_b, width, color='y', bottom=bar_a, label='B')
p3 = plt.bar(xloc, bar_c, width, color='b', bottom=bar_a+bar_b, label='C')

plt.xticks(xloc+width/2., (str(num+1) for num in xloc)) #dynamic instead of ('1', '2', '3', '4', '5', '6', '7', '8', '9', '10'))
plt.yticks(np.arange(0,300,10))
plt.ylabel('Scores')
plt.title('Index')
leg = plt.legend()
plt.grid(True)

#For setting and selecting legend items
barcontainers = [p1, p2, p3] #List of BarContainers
legendentry = [] #Mathing Lenged Entry to bar container

#Set pickers on the legend
for legenditem in leg.get_patches():
    legenditem.set_picker(5)
    legendentry.append(legenditem) #Append to legend entry to a list so we can get the index later

#Function to redraw on picking
def onpick(event):
    tempbarcontainers = barcontainers[:] #Make a copy of barcontainers. Used for re-ordering of bars
    selectedforbottom = legendentry.index(event.artist) #Index of selected artist

    #Use the index of the selected Legend entry as the index of the barcontainer list.
    bottombars = tempbarcontainers[selectedforbottom].patches #Bottom bar will be the one selected
    tempbarcontainers.pop(selectedforbottom) #Pop the selected artist from the list
    midbars = tempbarcontainers[0].patches #Mid bars is first of remaining bars
    topbars = tempbarcontainers[1].patches #Top bar is second of remaining bars

    for bottom, mid, top in zip(bottombars,midbars,topbars):
        bottom.set_y(0) #Set the bottom of each selected bar to 0
        mid.set_y(bottom.get_height()) #Set the bottom of the second bar to the top of bottom bar
        top.set_y(bottom.get_height() + mid.get_height()) #Set the bottom of third bar to the top of the sum of other bars

    fig.canvas.draw() #Redraw the canvas

fig.canvas.mpl_connect('pick_event', onpick)

plt.show()