python - statsmodels马赛克图-如何订购类别

这是数据帧：

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。有办法实现吗？

最佳答案

使用sort_values对大小列中存在的值进行排序以更改顺序。

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

关于python - statsmodels马赛克图-如何订购类别，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/39796183/