matplotlib: torne as teclas de legenda quadradas

Estou trabalhando com o matplotlib e gostaria de mudar as teclas das minhas legendas para serem quadrados em vez de retângulos quando faço, por exemplo, gráficos de barras. Existe uma maneira de especificar isso?

O que eu tenho agora:

O que eu quero:

Obrigado!