Predictive Data Analysis with Python/

...

/

Solution: Visualizing Seaborn Datasets

Solution: Visualizing Seaborn Datasets

This lesson gives a detailed review of the solutions to the challenges in the previous lesson.

We'll cover the following...

Heatmap solution #

Press + to interact
import pandas as pd
import seaborn as sns   
df = sns.load_dataset('flights')  # Reading flights dataset from seaborn package
# Reshaping the DataFrame
df = pd.pivot_table(df, values = 'passengers', index = ['month'], columns = 'year')
# Plotting the heatmap
sns.heatmap(df, annot = True, fmt = '')

The above problem plots the data on a heatmap after reshaping it.

On line 7, the pivot_table function is used to ...