target_column = "target"
# 特征选择步骤
#correlations = data.corr()[target_column].abs().sort_values(ascending=False) 带有排序
#correlations = data.corr()[target_column].abs()
# 选择与目标变量高度相关的特征,假设我们选择与目标变量相关性大于0.2的特征
correlation_matrix = data.corr()
target_correlation = correlation_matrix['target'].drop('target') # 删除目标变量自身的相关性
selected_features = correlations[correlations > 0.2].index.tolist()
print("Selected Features:", selected_features)
# 创建特征矩阵 X 和目标变量 y
X = data[selected_features]
#X = data.iloc[:,0:15]
y = data[target_column]
print(X)
# 绘图大小
plt.figure(figsize=(15,15))
sns.heatmap(X.corr(),annot=True,fmt='.2f',square = True) # 显示数值,设定显示保留几位小数(f为小数,.2为保留两位),设为正方形
plt.show()
结果图出问题,只有第一行有显示数值(见下图),请问是哪里错了,怎么修改?