简化值的 Python 数组检索

Question

我现在有以下代码可以完美运行：-

my_array = [
    ['var1', 1], ['var2', 2], ['var3', 3], ['var4', 4], ['var5', 5]
]

for i in range(len(my_array)):
    if my_array[i][0] == "var1":
        var_a = my_array[i][1]
    elif my_array[i][0] == "var2":
        var_b = my_array[i][1]
    elif my_array[i][0] == "var3":
        var_c = my_array[i][1]
    elif my_array[i][0] == "var4":
        var_d = my_array[i][1]
    elif my_array[i][0] == "var5":
        var_e = my_array[i][1]
        
print(var_a)
print(var_b)
print(var_c)
print(var_d)
print(var_e)

有没有一种方法可以简化获取值的方式，而不是使用多个“elif”？

我试过这样的事情： -

var_f = my_array[i]["var1"]
print(var_f)

但我收到一个错误：-

TypeError: list indices must be integer, not str

任何帮助将不胜感激！

谢谢

Answer 1

您可以将 my_array 转换为 dict 以简化值的检索：

my_array = [["var1", 1], ["var2", 2], ["var3", 3], ["var4", 4], ["var5", 5]]

dct = dict(my_array)

# print var1
print(dct["var1"])

印刷：

1