pandas scatter matrix display correlation coefficient

后端 未结 1 1787
后悔当初
后悔当初 2021-02-09 15:10

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.

1条回答
  •  心在旅途
    2021-02-09 15:47

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

    http://i.stack.imgur.com/apXPu.png

    0 讨论(0)
提交回复
热议问题