text_html_output_1.png
import matplotlib.pyplot as plt

import pandas as pd

import seaborn as sns

train_df = pd.read_csv('/kaggle/input/playground-series-s3e1/train.csv')
test_df = pd.read_csv('/kaggle/input/playground-series-s3e1/test.csv')

train_df.shape

train_df.columns

train_df.isnull().sum()

train_df = train_df.drop(['AveRooms'], axis=1)

train_df = train_df.drop(['AveBedrms'], axis=1)

plt.figure(figsize=(12, 10))
sns.heatmap(train_df.corr(), annot=True, cmap='Greens')