statsmodels mosaic plot - how to order categories

Question

Here is dataframe:

import pandas as pd
from statsmodels.graphics.mosaicplot import mosaic

df = pd.DataFrame({'size' : ['small', 'large', 'large', 'small', 'large', 'small'],
'length' : ['long', 'short', 'short', 'long', 'long', 'short']})

if I plot it mosaic(df, ['size', 'length']) it will display size in this order small then large, while I would like to have large and then small. Is there way to achieve it?

Just sort the `size` column: `mosaic(df.sort_values('size'), ['size', 'length'])` — Nickil Maveli, Sep 30 '16 at 16:40
@NickilMaveli very smart approach! You can post it as an official answer, if you like. — user1700890, Sep 30 '16 at 16:46

score 4 · Accepted Answer · answered Sep 30 '16 at 16:51

4

Use sort_values to sort values present in the size column to alter the order.

mosaic(df.sort_values('size'), ['size', 'length'])

answered Sep 30 '16 at 16:51

Nickil Maveli

29,155
8
82
85

Great! But now what about the `length` order...? :-( – Michel de Ruiter May 10 '23 at 09:36

statsmodels mosaic plot - how to order categories

1 Answers1