Потоки миграции между штатами
В видео вы видели тепловую карту потоков миграции между штатами, однако она получилась перегруженной. В этом упражнении вы рассмотрите только потоки внутри Среднего Запада.
DataFrame state_to_state уже загружен, а первые несколько строк отображаются в консоли. Напомним из видео: метки строк обозначают штат, в который переехали, а названия столбцов — штат, из которого переехали.
Список midwest_states содержит названия штатов Среднего Запада. (Выведите его в консоль, чтобы посмотреть, какие штаты в него входят.) DataFrame также использует названия штатов в качестве имён столбцов и индексов, поэтому для извлечения нужных столбцов и строк вы будете использовать midwest_states.
pandas и seaborn импортированы с обычными псевдонимами.
Это упражнение является частью курса
Анализ данных переписи населения США в Python
Инструкции к упражнению
- Отфильтруйте DataFrame, оставив только те столбцы, имена которых входят в список названий штатов, и только те строки, индексы которых также входят в этот список.
- Фильтрация могла изменить порядок столбцов и строк. Проверьте, совпадает ли
midwest.indexсmidwest.columns. - Отсортируйте DataFrame по индексу строк (
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()