CODEWITHSUNDEEP
pythonpandasplotvisualizationfacet
collarblind
collarblind

Reputation: 4739

How do i plot facet plots in pandas

This is what I have right now:

np.random.seed(1234)
test = pd.DataFrame({'week': [1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2],
                     'score': np.random.uniform(0, 1, 12),
                     'type': [0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1],
                     'type2': [3, 3, 4, 4, 5, 5, 3, 3, 4, 4, 5, 5]})

test.groupby(['week', 'type', 'type2']).agg('sum').unstack().plot(kind='bar')

enter image description here

How do I plot facet based on 'type'? I want two different plots, one for type = 1 and another type = 2.

Upvotes: 13

Views: 16288

Answers (1)

JAB
JAB

Reputation: 12781

You need to unstack so type are columns, and then use the subplots parameter:

test.groupby(['week', 'type', 
              'type2']).agg('sum').unstack(1).plot(kind='bar', subplots=True)

Resulting plot

Upvotes: 17

Related Questions