python input関数の再入力

Question

前提・実現したいこと

input関数で、指定した文字列以外の場合が入力された場合、再入力を指定したい。
20以上の入力項目をリスト化したい。
その20以上の要素は各番号が振られているので、入力はループ処理で対応したい。

発生している問題・エラーメッセージ

input関数に代入する要素が、ループに反映できません。
＊表示されるのは、常に1番目の要素になってしまいます。

該当のソースコード

#配列の内容を仮作成する
A = [1, 2, 3]
B = [11, 12, 13]

num = 0

while num in range(3):
    A_input = input('Aの1番目の要素を記入してください：')
    B_input = input('Bの1番目の要素の種類を記入してください：')
    if A_input == 'X' or A_input == 'Y':
        if B_input == 'Z' or B_input == 'V':
           A[num] = B_input
           B[num] = A_input
           num += 1
        else:
            print('Bの1番目の要素にはZ、Vのいずれかを記入してください。')
    else:
        print('Aの1番目の要素にはX、Yのいずれかを記入してください。')


print('入力した配列は', A, 'です。')
print('入力した配列は', B, 'です。')

python3

試したこと

forやwhileでループ表記しようとしましたが、input関数の中身が文字列表記のためループの内容に対応できませんした。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Comments

[y_waiwai]

このままではコードが読めないので、質問を編集し、<code>ボタンを押し、出てくる’’’の枠の中にコードを貼り付けてください

[s_kj]

y_waiwai様
コードを記載しました。
可能でしたら、コメントお願いいたします。

Answer 1

whileなどで、input関数以降をループの中に入れ、正常に入力させた場合にループを抜けるような処理を書きましょう

＃コードが読めないのでここまで

Comments

[s_kj]

ご回答ありがとうございます。
whileを使用すればいいということがわかることで、大変助かります。
そもそもどの構文が適していたのかもわかりませんでしたので。。。

Answer 2

コードを見ると、先頭と最後の部分と真ん中の処理が食い違っているのでなにをしたいのか今一つわからないのですが、真ん中の処理のところだけ見て、沢山入れなければならないとしたらこう作るだろうというものを書いてみました。

他にもいろいろなやりかたがあると思いますよ。
参考になれば。

def check_a(inp):
    if inp in ['X', 'Y']:
        return True, None
    else:  
        return False, 'Aの1番目の要素にはX、Yのいずれかを記入してください。'


def check_b(inp):
    if inp in ['Z', 'V']:
        return True, None
    else:  
        return False, 'Bの1番目の要素にはZ、Vのいずれかを記入してください。'


input_list = [
    {'value': None,
     'message': 'Aの1番目の要素を記入してください：',
     'check_func': check_a },
    {'value': None,
     'message': 'Bの1番目の要素を記入してください：',
     'check_func': check_b }
]

for item in input_list:
    res = False
    while res == False:
        user_input = input(item['message'])
        result, err_msg = item['check_func'](user_input)
        if result:
            item['value'] = user_input
            break
        else:
            print(err_msg)

値のチェック用にそれぞれ関数を作っていますが、もしこの可能な値をリストで判定する方式のものしかないのであれば、リストとエラーメッセージを保持するようにすればいいので、処理まもっと簡単になります。