如何將帶有 await 的 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))
)