Построение графика неясного Python

Question

Я пытался решить интересную проблему, но мне было трудно найти правильное решение. Я пытаюсь построить столбцы тепловых карт, где каждый столбец может варьироваться по размеру строки. Структура данных, которую я в настоящее время имею, состоит из вложенного списка, где каждый список содержит различные значения теплоты для их точек. Я приложу изображение ниже, чтобы прояснить это.

В настоящее время мы в основном пытались заставить работать matplotlib, однако мы не смогли получить ни одного из результатов, которые мы хотим. Пожалуйста, дайте мне знать, если у вас есть идеи о том, какие шаги нам следует предпринять.

Спасибо

Answer 1

Я думаю, что основной стратегией c было бы преобразование исходного массива так, чтобы у вас был прямоугольный массив angular с отсутствующими значениями, закодированными как NaN. Затем вы можете просто использовать imshow для отображения карты тепла.

Я использовал np.pad для заполнения пропущенных значений

data = [[0,0,1],[1,0,5,6,0]]
N_cols = len(data)
N_lines = np.max([len(a) for a in data])
data2 = np.array([np.pad(np.array(a, dtype=float), (0,N_lines-len(a)), mode='constant', constant_values=np.nan) for a in data])

fig, ax = plt.subplots()
im = ax.imshow(data2.T, cmap='viridis')
plt.colorbar(im)