Gruppengrößen visualisieren

Das Predictor-Insight-Diagramm liefert Informationen über prädiktive Variablen. Jede Variable teilt die Population in mehrere Gruppen. Das Diagramm enthält eine Linie, die die durchschnittliche Zielinzidenz für jede Gruppe zeigt, und Balken, die die Gruppengrößen darstellen. In dieser Übung lernst du, wie du eine Funktion schreibst und anwendest, die ein Predictor-Insight-Diagramm basierend auf einer entsprechenden Tabelle zeichnet.

Diese Übung ist Teil des Kurses

Einführung in Predictive Analytics mit Python

Kurs anzeigen

Interaktive Übung

Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.

# The function to plot a predictor insight graph
def plot_pig(pig_table, variable):
    
    # Plot formatting
    plt.ylabel("Size", rotation=0, rotation_mode="anchor", ha="right")
    
    # Plot the bars with sizes 
    pig_table["____"].plot(kind="bar", width=0.5, color="lightgray", edgecolor="none") 
    
    # Plot the incidence line on secondary axis
    pig_table["____"].plot(secondary_y=True)
    
    # Plot formatting
    plt.xticks(np.arange(len(pig_table)), pig_table[variable])
    plt.xlim([-0.5, len(pig_table) - 0.5])
    plt.ylabel("Incidence", rotation=0, rotation_mode="anchor", ha="left")
    
    # Show the graph
    plt.show()

Code bearbeiten und ausführen