Removing an axis or both axes from a matplotlib plot