Можно ли раскрасить рисунок Matplotlib Imshow?
Я работаю с некоторыми 2D-данными, в данном случае со спектрограммой речи, и я хотел бы отметить диаризацию речи на рисунке, если это возможно. Я очень быстро смоделировал хороший способ визуального обозначения диаризации, используя схему цветового кодирования для каждого столбца вверху. Можно представить, что такое цветовое кодирование также должно привести к тому, что объект matplotlib.legend() будет сбоку.
Чтобы создать маркировку, я хотел бы использовать одномерный вектор меток классов (например, [0, 0, 0, 1, 1, 0, 0, 0, 2, 2, 0 и т. д.] для всех столбцов) .
Было бы круто, если бы легенда также использовала качественные цветовые карты.
Короче говоря, есть ли способ сделать это изначально в Matplotlib?
... Или просто добавьте линию или прямоугольник с координатами осей в y и координатами данных в x, используя смешанное преобразование: matplotlib.org/3.1.0/tutorials/advanced/…
Ответы 1
Чтобы конкретизировать комментарий @ImportanceOfBeingErnest, вот как я буду делать то, что вы ищете:
from mpl_toolkits.axes_grid1 import make_axes_locatable
# use of `make_axes_locatable` simplifies positioning the
# accessory axes
# Generate data, it would have been nice if you had provided
# these in your question BTW
Ncols, Nlines = 200,50
data = np.random.random(size=(Nlines,Ncols))
class_labels = np.zeros(shape=(Ncols,))
class_labels[50:100] = 1
class_labels[100:150] = 2
class_labels = class_labels.reshape((1,Ncols))
fig, ax = plt.subplots(1,1)
# create new axes on the right and on the top of the current axes.
divider = make_axes_locatable(ax)
class_ax = divider.append_axes("top", size=0.1, pad=0., sharex=ax)
cbar_ax = divider.append_axes("right", size=0.1, pad=0.1)
#plot sonogram
im = ax.imshow(data, cmap='viridis')
fig.colorbar(im, cax=cbar_ax) # sonogram colorbar
# plot diarization classes
class_ax.imshow(class_labels, aspect='auto', cmap='rainbow')
class_ax.set_axis_off()
Спасибо за кодирование! Я вообще не знал о функции полезности make_axes_locatable. Приятно знать!
Да просто поставить другое имшоу в подсюжет или топоры сверху.