rio.plot.show с цветовой шкалой?

Question

Как я могу добавить цветную панель после использования rio.plot.show? Я перепробовал несколько вещей, но получил различные ошибки

Вот один из способов, который я пробовал:

fig, ax = plt.subplots(figsize = (16, 16))

retted = rio.plot.show(ds, ax=ax, cmap='Greys_r')  

fig.colorbar(retted, ax=ax)
plt.title("Original")
plt.show()

В этом есть ошибка: AttributeError: 'AxesSubplot' object has no attribute 'get_array'