python - 压缩多个 if 语句

Question

现在我的 python 代码如下所示:

if imput == "con":
    print "%.6f" % (con_getinfo["balance"])
elif imput == "xra":
    print "%.6f" % (xra_getinfo["balance"])
elif imput == "neg":
    print "%.6f" % (neg_getinfo["balance"])
elif imput == "grf":
    print "%.6f" % (grf_getinfo["balance"])
elif imput == "ninja":
    print "%.6f" % (ninja_getinfo["balance"])

现在我想让它看起来不那么重复，像这样:

if imput == con or imput == xra or imput == neg or imput == grf or imput == ninja:

但我不知道如何为每个条件分配正确的 react 。

Answer 1

您可以将对函数的引用存储在字典中，然后它只是一个简单的查找:

response = {
    "con": con_getinfo,
    "neg": neg_getinfo,
    ...
}
print "%.6f" % (response[imput]["balance"])