如何根据数据框中的值绘制带有颜色的表格？

Question

让我们假设我已经从矩阵传输了此数据帧A：

df

我想将其绘制为一个表格，其中较高的值将被标为深色，而较小的值将被标为浅色。但我不确定是否可行。所有的绘图技术似乎都不包括带有颜色的表格。有任何想法吗？

谢谢

Answer 1

使用matplotlib.pyplot.subplots：

import matplotlib.pyplot as plt

fig, ax = plt.subplots()
ax.matshow(df, cmap=plt.cm.Greys)

输出：

Answer 2

这是另一种方式：

import matplotlib.pyplot as plt
import numpy as np

a = df.to_numpy()
fig, ax = plt.subplots()

for i in range(df.shape[0]):
    for j in range(df.shape[1]):
         text = ax.text(j, i, a[i, j],
                        ha="center", va="center", color="w")

ax.set_title("Your title")
plt.imshow(a, cmap='hot', interpolation='nearest')
plt.show()

---

有关更多详细信息，请检查this URL。