使用不平衡的行和列创建棋盘图

Question

我有一个类似于这种格式X = [[1,4,5], [34,70,1,5], [43,89,4,11], [22,76,4]]的数据集，其中元素列表的长度不相等。

我想创建一个4行4列的棋盘图，每个单位框的颜色栏对应于数字的值。在此数据集中，将丢失一些小方框（例如，第4列第1行）。

我如何使用matplotlib在python中绘制这个？ 谢谢

Answer 1

您可以使用seaborn库或matplotlib来生成热图。首先，将其转换为pandas数据帧以处理缺失值。

import pandas as pd
df = pd.DataFrame([[1,4,5],[34,70,1,5], [43,89,4,11],[22,76,4]])
%matplotlib inline
from matplotlib import pyplot as plt
import seaborn as sns
sns.heatmap(df)
plt.show()

结果看起来像这样。