corr.py

import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
import numpy as np


df = pd.read_csv("~/Downloads/ev_vs_petrol_dataset_v3.csv")
numeric_df = df.select_dtypes(include = ["number"])

corr_matrix = numeric_df.corr(method="pearson")
plt.figure(figsize=(14, 12))
sns.heatmap(corr_matrix, annot=True, fmt=".2f")
plt.title("Correlation Matrix Heatmap")
plt.show()