text_plain_output_1.png image_output_1.png
import matplotlib.pyplot as plt

import pandas as pd

import seaborn as sns

df = pd.read_csv('../input/szeged-weather/weatherHistory.csv')

df.isnull().sum()

df = df.dropna()
df.isnull().sum()

df['Formatted Date'] = df['Formatted Date'].str.split(' ').str[0].str.split('-').str[0]

df = df.rename(columns={'Formatted Date': 'Year'})

df = df.drop(['Loud Cover'], axis=1)

a = df[df['Year'] == '2016']
a.shape

b = df[df['Year'] == '2007']
b.shape

plt.figure(figsize=[20, 10])
sns.heatmap(b.corr(), cmap='coolwarm', annot=True)