Отдельные легенды для сюжетов в matplotlib - PullRequest
Новая
       48

Отдельные легенды для сюжетов в matplotlib

0 голосов
/ 26 апреля 2020

Используя matplotlib, у меня есть фигура с подзадачами. Каждый субплот является тепловой картой, созданной из отдельного массива numpy. Я хочу напечатать отдельную легенду для каждого сюжета. Как мне это сделать?

В каком-то смысле мое требование противоположно этому вопросу: matplotlib - Легенда в отдельном сюжете

Спасибо Йохану C за предложение. У меня есть два требования: одно касается добавления отдельной легенды к каждому подпункту, а также легенда должна соответствовать тепловой карте. При дальнейшем поиске похоже, что ключевое слово для получения решения - colorbar, и следующий код работает для меня. Он основан на:

цветовой шкале matplotlib в каждом подплощадке https://matplotlib.org/examples/pylab_examples/custom_cmap.html

import matplotlib.pyplot as plt
import numpy as np

fig, axs = plt.subplots(3)

for i_ax in range(3):
    i_arr = np.random.rand(2, 2)

    im = axs[i_ax].imshow(i_arr)
    fig.colorbar(im, ax=axs[i_ax])
    # axs[i_ax].legend(loc="right")

fig.savefig('mpl_scratch.png')
...