Pyplot - format label in math mode of a variable