python - 如何将具有二进制值的 DataFrame 绘制为颜色矩阵？

Question

我有一个类似于以下内容的 DataFrame:

Name V1 V2 V3
A    1  0  0
B    1  0  1
C    1  1  0

等等

我想将其可视化为一个带有标记行和列的二维矩阵，以及根据值是 1 还是 0 着色的单元格。

换句话说，我想做这样的事情:Conditional coloring of a binary data frame in R

但在 Python 中。

感谢任何帮助!

Answer 1

将 Wen 的评论充实为答案 - 使用 seaborn.heatmap:

import matplotlib.pyplot as plt
import seaborn as sns

sns.set()

df = df.set_index('Name')
df

      V1  V2  V3
Name            
A      1   0   0
B      1   0   1
C      1   1   0

sns.heatmap(df)
plt.show()