將函數的輸出返回到字典

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))) 

這給你完全一樣的整數索引到value的映射。但是，只是一個清單會給你完全相同的索引到值的映射：

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

（如果some_test_function已經返回list對象，只是刪除list()調用）。