【Python】リスト名のリストからinput関数で入力した文字rと一致するリスト名のリストを抽出したい

Question

以下のリスト構造において
yamada = [0,1,2,3]
tanaka = [2,4,3,2]
sato = [5,2,4,8]
nakama = [yamada,tanaka,sato]
select = []

入力した文字と一致する配列を抽出したい
name = input()
select = nakama[nakama.index(name)]

エラー内容
ValueError: 'yamada' is not in list

期待値
yamadaと入力したら
select = [0,1,2,3]　と代入される事
tanakaと入力したら
select = [2,4,3,2]　と代入される事

ご教授の程、宜しくお願い致します

Comments

[TakaiY]

'yamada = [0,1,2,3]' のときに'yamada'と入力されてこの変数を選択するような、変数名そのものを変数的に使うのはプログラミング的に筋が悪いのですが、こうしなければならない理由はありますか？

Answer 1

nakama = {"yamada": yamada,"tanaka": tanaka,"sato": sato}

ですね。

普通は、ファイルから読んだりすると思うので、yamada等の変数は作らず、直接nakamaに登録します。

# linesは実際にはファイルから読んだデータ
lines = """yamda,0,1,2,3
tanaka,2,4,3,2
sato,5,2,4,8""".split("\n")

nakama = {}
for line in lines:
    name, *data = line.split(",")
    nakama[name] = [int(x) for x in data]


name = input().strip()
select = nakama[name] 

print(select)

Answer 2

辞書を使いましょう。

nakama = {'yamada':[0,1,2,3],'tanaka':[2,4,3,2],'sato':[5,2,4,8]}
name = input()
select = nakama[name]

Answer 3

nakama = [yamada,tanaka,sato] は必要ありません。
以下でできます。

>>> yamada = [0,1,2,3]
>>> tanaka = [2,4,3,2]
>>> sato = [5,2,4,8]
>>> name = input()
yamada
>>> select = globals()[name]
>>> print(select)
[0, 1, 2, 3]