將函數的輸出返回到字典

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