Цветовая полоса — это полоса, которая проходит вдоль диаграммы Matplotlib и действует как легенда для цветов, отображаемых на диаграмме.
Matplotlib по умолчанию отображает цветные полосы в правой части диаграммы, но вы можете легко изменить это с помощью функций из набора инструментов Matplotlib AxesGrid .
В этом руководстве показано несколько примеров использования этих функций на практике.
Пример 1: Разместите цветовую полосу на правой стороне диаграммы
В следующем коде показано, как создать диаграмму Matplotlib и как разместить цветовую полосу в правой части диаграммы:
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import make_axes_locatable
#make this example reproducible
np.random.seed(1)
#create chart
fig, ax = plt.subplots()
im = ax.imshow (np.random.rand(15,15))
ax.set_xlabel('x-axis label')
#add color bar
fig. colorbar (im)
plt.show()
Пример 2: Разместите цветовую полосу под графиком
В следующем коде показано, как разместить цветную полосу под диаграммой Matplotlib:
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import make_axes_locatable
#make this example reproducible
np.random.seed(1)
#create chart
fig, ax = plt.subplots()
im = ax.imshow (np.random.rand(15,15))
ax.set_xlabel('x-axis label')
#add color bar below chart
divider = make_axes_locatable (ax)
cax = divider. new_vertical (size='5%', pad=0.6, pack_start= True )
fig.add_axes (cax)
fig. colorbar (im, cax=cax, orientation='horizontal')
plt.show()
Обратите внимание, что аргумент pad создает отступ между осью X диаграммы и цветовой полосой. Чем выше значение pad, тем дальше будет цветовая полоса от оси X.
Пример 3: Разместите цветовую полосу над диаграммой
В следующем коде показано, как разместить цветную полосу над диаграммой Matplotlib:
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import make_axes_locatable
#make this example reproducible
np.random.seed(1)
#create chart
fig, ax = plt.subplots()
im = ax.imshow (np.random.rand(15,15))
ax.set_xlabel('x-axis label')
#add color bar below chart
divider = make_axes_locatable (ax)
cax = divider. new_vertical (size='5%', pad=0.4)
fig.add_axes (cax)
fig. colorbar (im, cax=cax, orientation='horizontal')
plt.show()
Вы можете найти больше руководств по Matplotlib здесь .