matplotlib customize the legend to show squares instead of rectangles,plt.show()