Toky mezi státy
Ve videu jsi viděl/a heatmapu migračních toků mezi státy, ale byla nepřehledná. V tomto cvičení se zaměříš pouze na toky v rámci středozápadu.
DataFrame state_to_state je načtený a první několik řádků se zobrazuje v konzoli. Z videa si pamatuj, že popisky řádků označují stát, do kterého se lidé přestěhovali, zatímco názvy sloupců označují stát, ze kterého odešli.
Seznam midwest_states je definován s názvy středozápadních států. (Vytiskni ho do konzole, pokud chceš vidět, které státy obsahuje.) DataFrame také používá názvy států jako názvy sloupců a index, takže pomocí midwest_states vyberou sloupce a řádky, které chceš v heatmapě použít.
pandas a seaborn jsou importovány pod obvyklými aliasy.
Toto cvičení je součástí kurzu
Analýza dat amerického sčítání lidu v Pythonu
Pokyny k cvičení
- Vyfiltruj DataFrame tak, aby obsahoval pouze sloupce odpovídající seznamu názvů států a pouze řádky, jejichž indexy jsou v seznamu názvů států.
- Filtrování mohlo změnit pořadí sloupců a řádků. Zkontroluj, zda je
midwest.indexrovnomidwest.columns. - Seřaď DataFrame podle indexu řádků (
axis = 0) a podle názvů sloupců (axis = 1). V obou případech použijinplace = True. - Vytvoř heatmapu z
midwest. Použij barevnou škálu žlutá-zelená-modrá scmap="YlGnBu".
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
# 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()