pandas scatter matrix display correlation coefficient

Question

I've tried to find a way to display correlation coefficients in the lower or upper tri of a pandas scatter matrix - can someone point me in the right direction? Thank you.

Use the pandas pandas.corr() function? pandas.pydata.org/pandas-docs/stable/generated/… — alacy
– alacy, Commented Jan 4, 2015 at 23:15

Ohjeah · Accepted Answer · 2022-03-06 19:09:31Z

27

A working minimal example

import matplotlib.pyplot as plt
import pandas as pd
import numpy as np
from pandas.plotting import scatter_matrix
df = pd.DataFrame(np.random.randn(100, 4), columns=['a', 'b', 'c', 'd'])
axes = scatter_matrix(df, alpha=0.5, diagonal='kde')
corr = df.corr().to_numpy()
for i, j in zip(*plt.np.triu_indices_from(axes, k=1)):
    axes[i, j].annotate("%.3f" %corr[i,j], (0.8, 0.8), xycoords='axes fraction', ha='center', va='center')
plt.show()

edited Mar 6, 2022 at 19:09

answered May 13, 2015 at 12:59

Ohjeah

1,30718 silver badges25 bronze badges

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

3 Comments

Sean Nguyen Over a year ago

You can also specify the figure size: scatter_matrix(df, alpha=0.5, figsize=(15, 10), diagonal='kde')

scatter Over a year ago

anyone looking to plot the lower triangle numbers replace triu with tril docs.scipy.org/doc/numpy-1.15.0/reference/generated/…

Mr. T Over a year ago

as_matrix() is long deprecated. Use to_numpy() instead.

Collectives™ on Stack Overflow

pandas scatter matrix display correlation coefficient

1 Answer 1

3 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

3 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related