如何将包含等待的for循环转换为asyncio.gather()

Question

如何使用asyncio.gather和map编写以下代码？

        for i in range(len(data)):
            candlestick = data[i]
            candlesticks = data[0: i + 1]
            await strategy.execute(candlesticks, candlestick.startTime)

Answer 1

你可以这样做：

from asyncio import gather, create_task
tasks = []
for i in range(len(data)):
    candlestick = data[i]
    candlesticks = data[0: i + 1]
    tasks.append(create_task(strategy.execute(candlesticks, candlestick.startTime)))
results = await gather(*tasks, return_exceptions=False)

Answer 2

如果您想特别使用map()，可以这样做：

from asyncio import gather, create_task

await gather(
    *map(
        lambda i: create_task(
            strategy.execute(data[0: i + 1], data[i].startTime)
        ),
        range(len(data))
)