Flöden mellan delstater
I videon såg du en heatmap över migrationsflöden mellan delstater, men den var svåröverskådlig. I den här övningen fokuserar du enbart på flöden inom Mellanvästern.
En DataFrame state_to_state har laddats in och de första raderna visas i konsolen. Kom ihåg från videon att radetiketternas anger den delstat man flyttat till, medan kolumnnamnen anger den delstat man flyttat från.
En lista midwest_states har definierats med namnen på delstaterna i Mellanvästern. (Skriv ut den i konsolen om du vill se vilka delstater som ingår.) Eftersom DataFrame:en också använder delstatsnamn som kolumnnamn och index kan du använda midwest_states för att extrahera de kolumner och rader du vill använda i heatmapen.
pandas och seaborn importeras med de vanliga aliaserna.
Den här övningen är en del av kursen
Analys av amerikansk folkräkningsdata i Python
Övningsinstruktioner
- Filtrera DataFrame:en så att du bara behåller de kolumner vars namn finns i listan över delstater, och de rader vars index finns i listan.
- Filtreringen kan ha ändrat ordningen på kolumner och rader. Kontrollera om
midwest.indexär lika medmidwest.columns. - Sortera DataFrame:en efter radindex (
axis = 0) och kolumnnamn (axis = 1). Användinplace = Truei båda fallen. - Skapa en heatmap över
midwest. Använd en gul-grön-blå färgskala medcmap="YlGnBu".
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# Retain only rows and columns of Midwest states
midwest = state_to_state[____][state_to_state.index.isin(____)]
# Are rows and columns still in the same order?
print(____)
# Sort the rows (by index) and columns (by name)
midwest.sort_index(axis = ____, ____)
midwest.sort_index(axis = ____, ____)
# Create a heatmap of migration flows
____
plt.xticks(rotation=90)
plt.yticks(rotation=0)
plt.show()