python - 如何修改 Pandas 绘图集成？

Question

我正在尝试修改 scatter_matrix Pandas 上可用的图。

简单的用法是

获得了:

iris = datasets.load_iris()
df = pd.DataFrame(iris.data, columns=iris.feature_names)
pd.tools.plotting.scatter_matrix(df, diagonal='kde', grid=False)
plt.show()

我想做几个修改，其中:

• 设法关闭所有地 block 上的网格
• 将任何 y 标签 x 旋转 90 度
• 关闭勾号

有没有办法让我修改 pandas 的输出而不必重写我自己的散点图函数？从哪里开始添加不存在的选项、微调等？

谢谢!

Answer 1

pd.tools.plotting.scatter_matrix 返回它绘制的轴数组；左下边界轴对应于索引 [:,0] 和 [-1,:]。人们可以遍历这些元素并应用任何类型的修改。例如:

axs = pd.tools.plotting.scatter_matrix(df, diagonal='kde')

def wrap(txt, width=8):
    '''helper function to wrap text for long labels'''
    import textwrap
    return '\n'.join(textwrap.wrap(txt, width))

for ax in axs[:,0]: # the left boundary
    ax.grid('off', axis='both')
    ax.set_ylabel(wrap(ax.get_ylabel()), rotation=0, va='center', labelpad=20)
    ax.set_yticks([])

for ax in axs[-1,:]: # the lower boundary
    ax.grid('off', axis='both')
    ax.set_xlabel(wrap(ax.get_xlabel()), rotation=90)
    ax.set_xticks([])