Auto数据集相关矩阵热力图

import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
auto = pd.read_csv('https://liangdaima.com/static/data/statistics/Auto.csv')
auto['horsepower'] = pd.to_numeric(auto['horsepower'], errors='coerce')
auto = auto.dropna()
numeric_cols = auto.select_dtypes(include=['float64', 'int64']).columns
corr = auto[numeric_cols].corr()
plt.figure(figsize=(10, 8))
sns.heatmap(corr, annot=True, cmap='coolwarm', center=0)
plt.title('Auto数据集相关矩阵热力图')
plt.show()