Actualmente estoy generando mapas de clúster en seaborn y etiquetando los colores de la fila como se muestra a continuación.

matrix = pd.DataFrame(np.random.random_integers(0,1, size=(50,4)))
labels = np.random.random_integers(0,5, size=50)

lut = dict(zip(set(labels), sns.hls_palette(len(set(labels)), l=0.5, s=0.8)))
row_colors = pd.DataFrame(labels)[0].map(lut)

g=sns.clustermap(matrix, col_cluster=False, linewidths=0.1, cmap='coolwarm', row_colors=row_colors)
plt.show()

Tengo una segunda columna de anotación similar a los datos de las etiquetas que también me gustaría agregar a la gráfica. La API de Seaborn no admite agregar un segundorow_colors columna, lo cual está bien, pero estoy luchando por encontrar una solución usando matplotlib para agregar esta columna de anotación a la agrupación.

Si no puedo usar seaborn para hacer esto y tengo que generar todo esto manualmente usando matplotlib que estaría bien, tampoco puedo entenderlo.

¡Gracias por tu ayuda!