結果 : arithmetic progression python code