将函数的输出返回到字典

Question

我有一个可以返回可变数量输出的函数。我想把它保存到这样的字典中：

val_dict={}
idx = 0

for key in some_test_function(1,2,3):
    val_dict[idx] = key
    idx +=1

现在虽然这件事确实有效，但我想知道是否有一种更优雅的方式。有什么想法吗？

Answer 1

您可以使用enumerate() function在词典中添加索引：

val_dict = dict(enumerate(some_test_function(1, 2, 3)))

这为您提供了与值映射完全相同的整数索引。但是，只需一个列表就可以获得完全相同的索引到值映射：

val_list = list(some_test_function(1, 2, 3))

（如果some_test_function已经返回list个对象，则只需删除list()来电。