从列表列表绘制热图

问题描述

拥有包含(不同大小的)变量列表的列表：

res = [[0.01, 0.9, 0.46], [0.64, 0.24], [0.87, 0.99, 0.47, 0.75], ...]

(请注意，每个实际列表包含约3000个变量)

假设：

x = [0, 1, 2, ...]

有没有办法用y轴上的res值制作热图？

例如，这是以下列表之一的热图：

我只想为每个列表添加此选项，并将它们放在同一热图上

---

解决方案

我通过执行以下操作找到了解决办法：

y_ax = []
cpt = 0
for elem in res:
    y_ax.append([cpt]*len(elem))
    cpt += 1

res_flat = sum(res, [])
y_flat = sum(y_ax, [])