如何优化Pandas散点矩阵图可读性？能否为每列添加上下变量名？

可行，你可以通过手动修改matplotlib的轴对象来实现这个需求。以下是调整后的代码：

import pandas as pd
%matplotlib inline
import matplotlib.pyplot as plt
import seaborn as sns

# 读取数据
df = pd.read_csv('train.csv')
# 生成散点矩阵并获取轴对象数组
axes = pd.plotting.scatter_matrix(df, figsize=(10,10))

# 遍历轴对象，设置顶部和底部的x轴标签
for i in range(len(axes)):
    for j in range(len(axes[i])):
        # 给每列顶部的子图设置x标签
        if i == 0:
            axes[i][j].set_xlabel(df.columns[j])
            axes[i][j].xaxis.label.set_rotation(45)
            axes[i][j].xaxis.label.set_ha('right')
        # 保留并调整底部子图的x标签格式
        if i == len(axes)-1:
            axes[i][j].xaxis.label.set_rotation(45)
            axes[i][j].xaxis.label.set_ha('right')

# 自动调整布局，确保标签完整显示
plt.tight_layout()
plt.show()

关键说明： #

• pd.plotting.scatter_matrix会返回一个二维轴对象数组，我们可以遍历这个数组修改每个子图的标签配置。
• 针对第0行的所有子图（对应每列顶部）设置x轴标签，就能实现顶部显示变量名的效果。
• 旋转标签并调整对齐方式，是为了避免标签重叠，提升可读性。
• plt.tight_layout()会自动调整子图间距，确保所有标签都能完整展示。

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