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

时间:2016-09-30 16:35:11

标签: python pandas statsmodels mosaic

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

如果我绘制mosaic(df, ['size', 'length']),它将按此顺序显示size small然后large,而我希望large然后{{} 1}}。有没有办法实现它?

1 个答案:

答案 0 :(得分:2)

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

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

Image