Использование colorbar () matplotlib, связанного с осями

Question

Я хочу иметь быстрый контурный контур () с цветовой шкалой ().

Если у меня есть только один график, который я использую:

plt.contourf(X, Y, Z, 20, cmap='RdGy')
plt.colorbar();

Но если я использую оси, как мне поступить?Т.е.

fig, ax = plt.subplots(2)

ax[0].contourf(X, Y, Z, 20, cmap='RdBu');
ax[1].contourf(Y, X, Z, 20, cmap='RdBu');

ax[0].colorbar(); #does not work

Я хотел бы иметь что-то быстрее, чем определить разделитель вручную и т. Д., Так же просто, как plt.colorbar () в первом примере.

Answer 1

colorbar - метод рисунка.Он принимает ScalarMappable в качестве входных данных и, необязательно, оси, рядом с которыми размещена цветовая панель.

fig, ax = plt.subplots(2)

im = ax[0].contourf(X, Y, Z, 20, cmap='RdBu')
im2 = ax[1].contourf(Y, X, Z, 20, cmap='RdBu')

fig.colorbar(im, ax=ax[0])

Существует обсуждение вопроса о том, как сделать цветовую панель доступной в качестве метода осей в github .Не стесняйтесь вносить свой вклад.