将函数的输出返回到字典

Question

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

val_dict={}
idx = 0

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

现在，尽管这件事确实可行，但我想知道是否还有一种更优雅的方式来做到这一点。 有任何想法吗？

Answer 1

您可以使用enumerate()函数在字典中添加索引：

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

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

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

（如果some_test_function已经返回list对象，则删除list()调用）。