确定相关矩阵中的颜色代码

Question

我看到了如下代码，该代码在具有python 3.7.3版本的Jupyter上运行。我在前面的教程中看到，一列与其自身的相关性是1，因此在矩阵的那个单元格中看到了红色，但是现在我看到自己尝试一下，我发现它是黄色的。这是因为颜色代码发生了某些变化还是由于python版本的更改？如代码部分“ ax.matshow（corr）”中所示，这里似乎有所更改。有没有一种方法可以定义我自己的颜色代码，使其不与强相关（0 ----> 1）

相关

def plot_corr(df,size):
    '''
    Function plots graphical corelation matrix for each pair of column in dataframe
    Input: 
          df: pandas dataframe
          size: vertical and horizontal size of the plot
    Displays: 
            matrix of corelation between columns. Blue-cyan-yellow-red-darkred => less to more corelated
                                                  0-------------------->1
                                                  Expect a dark red line running from top left to bottom right
    '''

    corr=df.corr()  #data frame corelation funnction
    fig,ax= plt.subplots(figsize=(size,size))
    ax.matshow(corr)  #color code the matrix rectangles by corelation value
    plt.xticks(range(len(corr.columns)),corr.columns) #draw xticks mark

Answer 1

这可以使用Matplotlib Colormaps完成，可以将其添加到函数matshow中，并添加参数cmap=plt.get_cmap(name)。

例如如果您想使用红色的颜色表，例如Reds，请运行

ax.matshow(corr, cmap=plt.get_cmap('Reds'))