Python Create New Variable Using If Statement