python - 如何将变量设置为属性？

Question

尝试将变量设置为属性时出现错误。

parser = argparse.ArgumentParser()

parser.add_argument('--arch', action='store',
                dest='arch', default='alexnet',
                help='Store a simple value') 

args = parser.parse_args()

model = models.args.arch(pretrained=True)

我知道 models.args.arch 会产生错误，但是将变量设置为属性的语法应该是什么样的？我可以用 if 语句来做到这一点，但这将是很多代码，我想这可能在 1 行中。

Answer 1

您想要访问内部字典来更新:

model = models.__dict__[args.arch](pretrained=True)

或使用getattr:

 getattr(models, args.arch)(pretrained=True)