空の値を持つstatsmodelsのモザイクプロット

Question

statsmodels.graphics.mosaicplot.mosaic()で空のセルをキャストするモザイクプロットを作成したいとします。セルはそのサイズにかかわらず作成されるため、結果のプロットは醜いものに見えます。

import matplotlib.pyplot as plt 

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

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

print(df) # note that the 'short'-'small' combination is missing 

fig = plt.figure() 
ax = fig.add_subplot(111) 
mosaic(df, ax=ax) 

は「短い小さな」値のために空のセルをプロットを作成します：例

は、このセルのかに作成を回避するためのいずれかの方法はあります後でそれをプロットから削除しますか？

Answer 1

は、その値が0でラベルを除外するために無名関数を送る：

labels = lambda k: "\n".join(k) if df[k] != 0 else "" 
mosaic(df, ax=ax, labelizer=labels)