how to have legend python