Общая легенда для группы подзаговоров

Question

Как показать общую легенду для групп подзаговоров ( не все ). Например, одна легенда для всех вспомогательных участков в каждом столбце.

fig, ax = plt.subplots(nrows=4, ncols=2)

for row in range(0, 4):
    for col in range(0, 2):
        ax[row][col].legend(loc="upper right") 
        # This will add a legend for each sub plot.

fig.legend(loc="upper right")
# This will add a legend for whole figure. 
# Suggested here: https://stackoverflow.com/a/46921590/947889

Вместо этого мне нужна общая легенда для каждого столбца (или строки). Это может быть полезно, когда графики в разных столбцах различны (например, в моем случае графики в первом столбце представляют собой тепловую карту, а графики во втором столбце представляют собой линейные графики, которые, очевидно, должны иметь различную легенду).

Answer 1

Вы можете сделать что-то вроде этого:

fig, ax = plt.subplots(nrows=4, ncols=2)

for row in range(0, 4):
    for col in range(0, 2):
        if col == 1: # only for the last column, add a legend
            ax[row][col].legend(loc="upper right")