如何修改Python代码将相关性矩阵结果保留两位小数？

这里提供几种简单的修改方式，让输出的相关性结果保留两位小数：

方法1：直接对相关性序列做小数位处理 #

使用pandas的round()方法对目标序列进行四舍五入，保留两位小数：

corr_matrix = data.corr()
corr_matrix.sort_values('SalePrice', ascending=False, inplace=True)

# 保留两位小数后输出
print(corr_matrix.SalePrice.round(2))

方法2：自定义打印格式 #

通过格式化字符串控制输出的小数位数，适合需要自定义输出样式的场景：

corr_matrix = data.corr()
corr_matrix.sort_values('SalePrice', ascending=False, inplace=True)

# 遍历并格式化每个相关性值
for feature, corr_value in corr_matrix.SalePrice.items():
    print(f"{feature}: {corr_value:.2f}")

方法3：提前处理整个相关性矩阵 #

如果后续还会用到这个相关性矩阵，可以在计算时就统一保留两位小数：

# 计算相关性时直接保留两位小数
corr_matrix = data.corr().round(2)
corr_matrix.sort_values('SalePrice', ascending=False, inplace=True)

print(corr_matrix.SalePrice)

内容的提问来源于stack exchange，提问作者Abhishek Venkatesh