Python 可视化:热力图 + 相关性分析 + 特征筛选

import pandas as pdimport numpy as npdf = pd.DataFrame({    "age": np.random.randint(18,60,300),    "income": np.random.normal(8000,2500,300).round(2),    "consume": np.random.normal(3000,1000,300).round(2),    "active": np.random.randint(1,31,300),    "score": np.random.uniform(3.0,5.0,300).round(1)})df.to_excel("corr_data.xlsx", index=False)print(" corr_data.xlsx 生成完成")

import pandas as pdimport seaborn as snsimport matplotlib.pyplot as pltplt.rcParams["font.sans-serif"] = ["SimHei"]plt.rcParams["axes.unicode_minus"] = Falsedf = pd.read_excel("corr_data.xlsx")corr = df.corr()plt.figure(figsize=(10,6))sns.heatmap(corr, annot=True, cmap="coolwarm", fmt=".2f", linewidths=0.5)plt.title("特征相关性热力图")plt.tight_layout()plt.show()