wake-up-neo.com

Pandas Zeichnen mit Multi-Index

Nach dem Ausführen einer groupby.sum() auf einem DataFrame habe ich einige Probleme beim Versuch, mein beabsichtigtes Diagramm zu erstellen.

grouped dataframe with multi-index

Wie kann ich eine Nebenhandlung erstellen (kind='bar') für jedes Code, wobei die x-Achse die Month ist und die Balken ColA und ColB sind?

54
Reustonium

Ich fand, dass die Unstack-Methode (Level-Methode) perfekt funktioniert, was den zusätzlichen Vorteil hat, dass keine Vorkenntnisse über die Anzahl der Codes erforderlich sind.

df.unstack(level=0).plot(kind='bar', subplots=True)
83
Reustonium

Mit dem folgenden DataFrame ...

DataFrame

# using pandas version 0.14.1
from pandas import DataFrame
import pandas as pd
import matplotlib.pyplot as plt

data = {'ColB': {('A', 4): 3.0,
('C', 2): 0.0,
('B', 4): 51.0,
('B', 1): 0.0,
('C', 3): 0.0,
('B', 2): 7.0,
('Code', 'Month'): '',
('A', 3): 5.0,
('C', 1): 0.0,
('C', 4): 0.0,
('B', 3): 12.0},
'ColA': {('A', 4): 66.0,
('C', 2): 5.0,
('B', 4): 125.0,
('B', 1): 5.0,
('C', 3): 41.0,
('B', 2): 52.0,
('Code', 'Month'): '',
('A', 3): 22.0,
('C', 1): 14.0,
('C', 4): 51.0,
('B', 3): 122.0}}

df = DataFrame(data)

... können Sie Folgendes zeichnen (mit Querschnitt):

f, a = plt.subplots(3,1)
df.xs('A').plot(kind='bar',ax=a[0])
df.xs('B').plot(kind='bar',ax=a[1])
df.xs('C').plot(kind='bar',ax=a[2])

enter image description here

Einer für A, einer für B und einer für C, x-Achse: 'Monat', die Balken sind ColA und ColB. Vielleicht ist es das, wonach Sie suchen.

15