Creating Array In Python Using And In If Statement