Pandas Datetime: создайте тепловую карту для сравнения первых 10 лет, в течение которых наблюдался НЛО, и каждого месяца.

Pandas Datetime: упражнение 24 с решением

Напишите программу Pandas для создания тепловой карты (прямоугольные данные в виде матрицы с цветовой кодировкой) для сравнения первых 10 лет, в течение которых наблюдался НЛО, и каждого месяца.

Пример решения:

Код Python:

import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
#Source: https://bit.ly/1l9yjm9
df = pd.read_csv(r'ufo.csv')
df['Date_time'] = df['Date_time'].astype('datetime64[ns]')
most_sightings_years = df['Date_time'].dt.year.value_counts().head(10)
def is_top_years(year):
   if year in most_sightings_years.index:
       return year
month_vs_year = df.pivot_table(columns=df['Date_time'].dt.month,index=df['Date_time'].dt.year.apply(is_top_years),aggfunc='count',values='city')
month_vs_year.columns = month_vs_year.columns.astype(int)
print("\nHeatmap for comparison of the top 10 years in which the UFO was sighted vs each month:")
plt.figure(figsize=(10,8))
ax = sns.heatmap(month_vs_year, vmin=0, vmax=4)
ax.set_xlabel('Month').set_size(20)
ax.set_ylabel('Year').set_size(20)

Пример вывода:

 Тепловая карта для сравнения 10 лучших лет, в которые НЛО наблюдался с каждым месяцем:

Редактор кода Python:

Есть другой способ решить это решение? Внесите свой код (и комментарии) через Disqus.

Предыдущий: Напишите программу Pandas, чтобы создать сравнение первых 10 лет, в течение которых был обнаружен НЛО, и каждый месяц.
Далее: Напишите программу Pandas, чтобы создать сравнение времени-года с годами для сравнения первых 10 лет, в которые наблюдался НЛО.