Formatting your Python Plot