如何更改列表取决于字符串值？

Question

我想根据输入字符串打印列表

代码如下：

A1 = [1,2]
B1 = [2,3]

p = input("A1/B1 : ")

print(p)
#if input is "A1" it will print A1 list or "A2" it will print A2

Answer 1

是的，您可以将数据存储在dict中并获取它来检索您想要的数据：

d = {'A1': [1,2], 'B1': [2,3]}

p = input("A1/B1 : ")

element = d.get(p, None)
if element is None:
    print("Sorry, not found")
else:
    print(element)

Answer 2

你在找类似的东西吗？

if p == 'A1':
  print(A1)
elif p == 'A2':
  print(A2)
else:
  print('Your answer must be A1 or A1')