if：条件NG時の再計算

Question

###質問事項
質問タイトルがわかり辛くすみません。

pythonの基礎勉強をしています。
下記補足に記したように、whileループの中にif文を2つ作り、各々ifの条件次第で、各々のif文の頭に戻るようにしたいのですが、良いやり方がわかりません。

１つ目のif文に関してはcontinueによりwhileの次のループに行けば良いと考えたのですが、２つ目のif文ではどのように記載すれば良いか分からず、補足に記したように、２つ目のif文を（無意味に）whileループで囲い、こちらもcontinueを用いました。

結果、期待通りに動作するのですが、もっとsmartな書き方があるのだろうと想像しています。
何か、スッキリした書き方はないでしょうか？
すみませんが、ご教示の程、よろしくお願いします。

##補足

i = 1  
out = []
while i>0:
    print('i= '+str(i))
    ##if(1)
    in1 = input('go or exit: ')
    if in1 == 'go':
        i += 1
    elif in1 == 'exit':
        break
    else:
        print('Invalid input.')
        continue
    ##if(2)
    while i>0:
        in2 = input('Enter a or b: ')
        if in2 == 'a':
            out.append('a')
            break
        elif in2 == 'b':
            out.append('b')
            break
        else:
            print('Invalid input.')
            continue
    print ('out= ',out)

Answer 1

スマートに見えないのは、メインループと入力ループが同じ場所にあるからかと思います。
以下の様に入力ループを関数に切り出してあげると幾分読みやすいコードになるかと思います。

def input_item(ask, cand):
  while True:
    inp = raw_input(ask)
    # 候補にある場合だけ結果を返す
    if inp in cand:
      return inp
    print('Invalid input.')

def main():
  i = 1
  out = []
  while True:
    print('i= '+str(i))
    inp = input_item('go or exit: ', ('go', 'exit'))
    if inp == 'exit':
      break
    i += 1
    inp = input_item('Enter a or b: ', ('a', 'b'))
    out.append(inp)
    print ('out= ',out)
  return out

main()

Comments

[motto]

入力ループを切り出すと、メインがスッキリしますね！メインも読みやすくなりました。ありがとうございました！

Answer 2

入力の検証を別のメソッドに切り分けて、その中でループさせるようにしてください。
以下の例では再帰を用いて、正しい結果が返るまでループさせています。

def validate_input(msg, ls):
    inp = input(msg)
    if inp in ls:
        return inp
    else:
        return validate_input(msg, ls)

i = 1  
out = []
while i>0:
    print('i= '+str(i))
    ##if(1)
    in1 = validate_input('go or exit: ', ('go','exit'))
    if in1 == 'go':
        i += 1
    elif in1 == 'exit':
        break
    ##if(2)
    in2 = validate_input('Enter a or b: ', ('a', 'b'))
    if in2 == 'a':
        out.append('a')
    elif in2 == 'b':
        out.append('b')
    print ('out= ',out)

Comments

[chelsy7110]

validate_inputの中のvalidate_input(msg, ls)にはreturnをつけないとおかしな動作になると思います。
また、この場合は再帰を使うのはあまり適当でないのでは無いでしょうか。
validate_inputの中のelse以下を削除して、 inp in ls がtrueとなるまでvalidate_inputメソッドの中でループさせる方がスマートでしょう。
あと、in2の方はbreakいらないと思います。

[tsuemura]

ご指摘ありがとうございます。修正しました。

> また、この場合は再帰を使うのはあまり適当でないのでは無いでしょうか。
> validate_inputの中のelse以下を削除して、 inp in ls がtrueとなるまでvalidate_inputメソッドの中でループさせる方がスマートでしょう。
これを言われてしまうと、僕は別途「ループと再帰どっちがスマートですか？」という質問を立てなくてはいけなくなってしまいます（ｗ
ただ、本質はメソッドの切り出しにあるので、再帰をことさらに強調する必要はないですね。説明文に補記しました

[motto]

＞TakuyaSuemuraさん
入力検証を切り出すことでスッキリしました！こういう場面でも、再帰が使えるのですね、勉強になりました。どうもありがとうございました。
＞chelsyさん
追加コメントありがとうございました。