python - 如何在python中使用变量作为字典的键

Question

嗨，我是 python 的新手，所以如果我问的是非常愚蠢的问题，请原谅。所以，我有一个名为 array 的字典，它有一些值，现在我想将键作为变量输入并使用它来打印分配给该键的值。

array = {'color':'blue' , 'size':'small'}
print array ['color']

这工作得很好，输出值蓝色。但如果我尝试这样做，它就不起作用。

array = {'color':'blue' , 'size':'small'}
var = input ('input a key') #input would be " color " or " size "
print array[var]

我认为有一个非常简单的解决方案。感谢您提前提供帮助:)

Answer 1

您的输入包含前导和尾随空格。例如，'color' 与 'color' 不同。您的字典没有关键字 ' color '。您可以像这样从输入字符串中去除空格:

>>> mydict = {'color':'blue' , 'size':'small'}
>>> var = input().strip()
' color '
>>> mydict[var]
'blue'

顺便说一下，您的array 是一个Python 字典，称它为array 有点令人困惑。