Plotta gruppstorlekarna
Prediktorinsiktsgrafen ger information om prediktiva variabler. Varje variabel delar upp populationen i flera grupper. Grafen innehåller en linje som visar den genomsnittliga målincidensen för varje grupp, samt ett stapeldiagram som visar gruppstorlekarna. I den här övningen lär du dig att skriva och använda en funktion som plottar en prediktorinsiktsgraf utifrån en prediktorinsiktsgraftabell.
Den här övningen är en del av kursen
Introduktion till prediktiv analys i Python
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# 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()