statsmodels 'Mosaik-Plot mit leeren Werten

Question

Ich möchte ein Mosaik-Plot mit statsmodels.graphics.mosaicplot.mosaic() mit Daten erstellen, die leere Zellen wirft. Sie erscheinen in der resultierenden Darstellung hässlich, weil eine Zelle unabhängig von ihrer Größe erstellt wird.

Beispiel:

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) 

für Wert „short kleinen“ ein Grundstück mit einer leeren Zelle erzeugt:

Gibt es eine Möglichkeit, die Schaffung dieser Zelle entweder zu vermeiden oder zu Entferne es danach aus der Handlung?

Answer 1

eine anonyme Funktion senden Etiketten, deren Werte 0 auszuschließen:

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