python - 当我从同一个函数中获取键和值时，使用字典理解构建一个字典

Question

假设我有一个复杂的函数get_stuff，它接受一个 int 并返回一个元组，第一个元素是 str 类型。以下示例具有相同的行为，但假设实际功能更复杂且不能轻易一分为二：

def get_stuff(x):
    return str(5*x),float(3*x)

我想要的是构建一个字典，其 (key,value) 对是 get_stuff 在特定整数集上调用时的结果。一种方法是：

def get_the_dict(set_of_integers):
    result = {}
    for i in set_of_integers:
        k,v = get_stuff(i)
        result[k] = v
    return result

我宁愿为此使用dict理解，但我不知道是否可以在理解中拆分对以分别捕获键和值。

def get_the_dict_with_comprehension(set_of_integers):
    return {get_stuff(i) for i in set_of_integers} #of course this doesn't work

我怎样才能做到这一点？

标签： pythondictionary-comprehension

Answer 1

你可以这样做，而不是字典理解：

dict(get_stuff(i) for i in range(10))