如何在异步收集中获得以下输出

Question

有没有办法我可以得到 result1 和 result2 的 as_completed 结果 合并？

预期输出

[0, 0, 1, 1, 2, 2]

电流输出

[0, 1, 2, 0, 1, 2]

代码

import asyncio

async def sleep(i):
    await asyncio.sleep(i)
    return i

async def main():
    result1 = [sleep(i) for i in [2, 1, 0]]
    result2 = [sleep(i) for i in [2, 1, 0]]

    result1Gathered = asyncio.as_completed(result1)
    result2Gathered = asyncio.as_completed(result2)

    result = await asyncio.gather(*result1Gathered, *result2Gathered)

    print(result)

loop = asyncio.get_event_loop()
loop.run_until_complete(main())

Answer 1

这是因为gather 返回给它的结果。

您想要的不是gather，而是循环遍历as_completed 的结果。您可以使用itertools.chain 组合所有可迭代对象：

from itertools import chain

...

async def main():
    result1 = [sleep(i) for i in [2, 1, 0]]
    result2 = [sleep(i) for i in [2, 1, 0]]

    result = [await i for i in asyncio.as_completed(chain(result1, result2))]

    print(result)

结果：

[0, 0, 1, 1, 2, 2]