Как получить доступ к свойствам цветовой шкалы через экземпляр осей?

Question

Я использую функцию, которая использует MatPlotLib для создания графика и связанной цветовой шкалы. Однако функция возвращает только экземпляр осей, а не цветной столбец.

Я хотел бы изменить метку цветного столбца. Есть ли способ доступа к свойствам цветовой шкалы только для экземпляра осей?

Answer 1

Основным принципом дизайна c является то, что функция должна возвращать художника, которого она создала. От этого вы можете получить цветную полосу:

import matplotlib.pyplot as plt

def function(ax=None):
    ax=ax or plt.gca()
    sc = ax.scatter([1,2,3], [1,3,2])
    ax.figure.colorbar(sc)
    return sc

fig, ax = plt.subplots()
sc = function(ax=ax)

sc.colorbar.set_label("My Label")

plt.show()