Python3: refactor ':FOR' statements