statsmodels de mosaique - comment commander catégories

Question

Voici 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']}) 

si je tracer le graphique mosaic(df, ['size', 'length']) il affichera size dans cet ordre small puis large, alors que je voudrais avoir large puis small. Y a-t-il moyen de l'atteindre?

Answer 1

Utilisez sort_values pour trier les valeurs présentes dans la colonne de taille afin de modifier l'ordre.

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