Как вывести более одной тепловой карты и иметь описание для цветовой шкалы

Question

У меня есть следующий код:

import numpy as np; np.random.seed(0)
import seaborn as sns; sns.set()

uniform_data1 = np.random.rand(10, 12)
ax1 = sns.heatmap(uniform_data1)

uniform_data2 = np.random.rand(5, 6)
ax2 = sns.heatmap(uniform_data2)

, который выводит своего рода «комбинированную» тепловую карту с «комбинированной» цветовой шкалой.

Как вывести 1-ю тепловую карту, за которой следует 2-я тепловая карта (расположена ниже 1-й), если мне нужно, чтобы весь мой код находился в одной ячейке, как показано выше? Мне это нужно, потому что я создаю отчет.

Кроме того, как мне получить описание для цветовой панели?

Заранее большое спасибо.

Answer 1

как то так?

import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt

fig, axes = plt.subplots(2, 1)

np.random.seed(0)

uniform_data1 = np.random.rand(10, 12)
ax1 = sns.heatmap(uniform_data1, ax=axes[0])

uniform_data2 = np.random.rand(5, 6)
ax2 = sns.heatmap(uniform_data2, ax=axes[1])

plt.show()

Редактировать Для описания, может быть с axes.legend()?