如何用二进制值作为颜色矩阵绘制数据帧？

Question

我有一个类似于以下的数据帧：

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

将温的评论充实成一个答案-使用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

^{pr2}$

Answer 2

可以将它们绘制为具有各种cmap替代项的图像：https://matplotlib.org/examples/color/colormaps_reference.html

但是从这里你需要更多的格式。。。在

import matplotlib.pyplot as plt
plt.imshow(df.values, interpolation="nearest", cmap='Blues')
plt.show()