Create Array Using For Loop Python Questions For If Statement