wzy-1642 2024-05-09 10:24 采纳率: 16.7%
浏览 166

相关性可视化热力图不显示数值


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()

结果图出问题,只有第一行有显示数值(见下图),请问是哪里错了,怎么修改?

img

  • 写回答

4条回答 默认 最新

  • 专家-皆知 2024-05-09 10:24
    关注

    引自免费微信小程序:皆我百晓生

    请确认你的代码中使用了正确的列名,并在绘制热力图时包含了数值数据。例如:sns.heatmap(data.corr(), annot=True)。这样会显示相关系数的数值。

    评论

报告相同问题?

问题事件

  • 创建了问题 5月9日