python - pcolormesh 上的标签

Question

我有以下数据框，我将其绘制为矩阵:

symbol      abc    def   xyz
symbol
abc         1     0.2    0.5
def         0.2     1    0.3
xyz         0.5   0.3      1

我将其绘制如下:

plt.pcolormesh(df.values)

有没有办法绘制单元格内每个单元格的值？换句话说，我不想在侧面有图例，我想将每对的值绘制在矩阵的相关单元格内。

Answer 1

我认为您需要为pcolormesh上的每个单元格创建一个plt.text对象。这个循环应该做你想做的事。可能有一种矢量化的方法可以做到这一点，但至少这是有效的。

for ii,i in enumerate(df.index):
    for jj,j in enumerate(df.keys()):
        plt.text(ii+0.5,jj+0.5,df[i][j])

这是完整的脚本:

import pandas as pd
import matplotlib.pyplot as plt

plt.viridis()

df = pd.DataFrame({
    'abc':{'abc':1.0,'def':0.2,'xyz':0.5},
    'def':{'abc':0.2,'def':1.0,'xyz':0.3},
    'xyz':{'abc':0.5,'def':0.3,'xyz':1.0}})

plt.pcolormesh(df.values)

for ii,i in enumerate(df.index):
    for jj,j in enumerate(df.keys()):
        plt.text(ii+0.5,jj+0.5,df[i][j],
                ha='center',va='center',
                size=20,color='w')

plt.show()