How to modify pandas plotting integration?

Question

I'm trying to modify the scatter_matrix plot available on Pandas.

Simple usage would be IRIS scatter matrix viz

Obtained doing :

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

I want to do several modification, among which:

managing to turn off grid on all plots
rotate x any y labels 90 degree
turn ticks off

Is there a way for me to modify pandas' output without having to rewrite my own scatter plot function ? where to start to add non-existing options, fine tunings, etc ?

Thanks !

behzad.nouri · Accepted Answer · 2014-04-11 11:14:42Z

7

pd.tools.plotting.scatter_matrix returns an array of the axes it draws; The lower left boundary axes corresponds to indices [:,0] and [-1,:]. One can loop over these elements and apply any sort of modifications. For example:

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

scatter

edited Apr 11, 2014 at 11:14

answered Apr 11, 2014 at 11:04

behzad.nouri

78.5k18 gold badges130 silver badges127 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How to modify pandas plotting integration?

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related