Python тепловая карта с неравными размерами блоков

Question

У меня есть следующий набор данных:

results=[array([6.06674849e-18, 2.28597646e-03]), array([0.02039694, 0.01245901, 0.01264321, 0.00963068]), array([2.28719585e-18, 5.14800709e-02, 2.90957713e-02, 0.00000000e+00,
       4.22761202e-19, 3.21765246e-02, 8.86959187e-03, 0.00000000e+00])]

Я хотел бы создать из него тепловую карту, которая выглядит аналогично следующему рисунку:

Можно ли создать такую ​​диаграмму с помощью seaborn или matplotlib или любого другого графического пакета, и если да, то как это сделать?

Answer 1

Одним из подходов является выравнивание длин строк с помощью np.repeat. Это хорошо работает, только если все строки имеют длину, которая является делителем самой длинной длины строки.

Данные предлагают использовать LogNorm , хотя такая норма отвлекается с нулями в Пример ввода.

Некоторый код, иллюстрирующий идею:

from matplotlib import pyplot as plt
from matplotlib import colors as mcolors
import numpy as np

results = [np.array([6.06674849e-18, 2.28597646e-03]),
           np.array([0.02039694, 0.01245901, 0.01264321, 0.00963068]),
           np.array([2.28719585e-18, 5.14800709e-02, 2.90957713e-02, 0.00000000e+00,
                     4.22761202e-19, 3.21765246e-02, 8.86959187e-03, 0.00000000e+00])]
longest = max([len(row) for row in results])
equalized = np.array( [np.repeat(row, longest // len(row)) for row in results])
# equalized = np.where(equalized == 0, np.NaN, equalized)
norm = mcolors.LogNorm()
heatmap = plt.imshow(equalized, cmap='nipy_spectral', norm=norm, interpolation='nearest',
                     origin='lower', extent=[0, 6000, 0.5, len(results)+0.5])
plt.colorbar(heatmap)
plt.gca().set_aspect('auto')
plt.yticks(range(1, len(results) + 1))
plt.show()

Другой пример с 7 уровнями (случайными числами). Входные данные генерируются как:

bands = 7
results = [np.random.uniform(0, 1, 2**i) for i in range(1, bands+1)]