matplotlib: hacer que las llaves de la leyenda sean cuadradas

Estoy trabajando con matplotlib y me gustaría cambiar las claves de mis leyendas para que sean cuadrados en lugar de rectángulos cuando hago, por ejemplo, gráficos de barras. ¿Hay alguna manera de especificar esto?

Lo que tengo ahora:

Lo que quiero:

¡Gracias!

Respuestas a la pregunta(3)

Su respuesta a la pregunta