Потоки між штатами
У відео ви бачили теплову мапу міграційних потоків між штатами, але вона була перевантажена. У цій вправі ви розглянете лише потоки в межах Середнього Заходу.
Датафрейм state_to_state уже завантажено, і перші кілька рядків показано в консолі. Пам'ятайте з відео: мітки рядків указують штат, до якого переїхали, а назви стовпців — штат, з якого виїхали.
Список midwest_states містить назви штатів Середнього Заходу. (Виведіть його в консоль, якщо хочете побачити, які штати входять до списку.) У датафреймі також використано назви штатів як назви стовпців та індекс, тож ви використаєте midwest_states, щоб вибрати потрібні стовпці та рядки для цієї теплової мапи.
pandas і seaborn імпортовано зі звичними псевдонімами.
Ця вправа є частиною курсу
Аналіз даних перепису США в Python
Інструкції до вправи
- Виберіть підмножину датафрейма: залиште лише ті стовпці, що відповідають списку назв штатів, і лише ті рядки, чиї індекси є в цьому списку.
- Підмножина могла змінити порядок стовпців і рядків. Перевірте, чи
midwest.indexдорівнюєmidwest.columns. - Відсортуйте датафрейм за індексом рядків (
axis = 0) і за назвами стовпців (axis = 1). У обох випадках використайтеinplace = True. - Створіть теплову мапу для
midwest. Застосуйте жовто-зелено-блакитну градацію кольорів за допомогоюcmap="YlGnBu".
Інтерактивна практична вправа
Спробуйте виконати цю вправу, доповнивши цей зразок коду.
# 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()