python asyncio.gather 使用输入作为返回值的一部分

Question

我有一个 function 之类的东西

async def funcn(input: input_type) -> output_type:
  ....
  return output

我使用 asyncio.gather 来调用它，例如：

output = await asyncio.gather(*[funcn(input) for input in input_list])

这里的返回值将是 input_list 中每个输入的 output 列表。

但我希望 input_list 中每个输入的返回值类似于 [(input, output)]。 在不改变 function 的实现的情况下是否有可能？

Answer 1

asyncio.gather的返回将以与输入相同的顺序返回，因此您只需将 zip 您的input_list与output一起返回：

for input_value, output_value in zip(input_list, output):
    print(input_value, output_value)

Answer 2

除了zip的回答中的 zip 之外，您始终可以使用另一个返回输入的 function 和 output 来包装funcn 。 这允许您在不修改处理 function 的情况下装饰 output：

async def wrap_funcn(input):
    output = await funcn(input)
    return input, output

...
ins_and_outs = await asyncio.gather(*[wrap_funcn(input) for input in input_list])

这在input_list是一个迭代器的情况下很有用，它的内容是在调用gather()之后耗尽它的。