statsmodels 马赛克图 - 如何排序类别

Question

这是数据框：

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

如果我绘制它 mosaic(df, ['size', 'length']) 它会按顺序显示 size small 然后 large，而我想显示 large 然后small。有什么办法可以实现吗？

Answer 1

使用 sort_values 对尺寸列中的值进行排序以更改顺序。

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