python - 将函数的输出返回到字典

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() 调用)。