python-3.x - 如何在 map 函数中使用 await

Question

我有以下代码

async def foo():
   some_tuple = tuple(map(bar, some_tuple))

async def bar(data):
   # Await another function and do some other stuff
   return something

因为 bar 是异步的，所以必须等待它。但是，我不确定在哪里等待 bar。我尝试在 map 中等待它，我尝试等待 map，我尝试等待 tuple，但没有任何效果。

如何在 map 中等待 bar？

Answer 1

您不能将 map 与异步函数一起使用，因为 map 是同步的并且不会在循环内暂停执行。这同样适用于需要同步生成器的 tuple 构造函数。

Ti 解决这个问题，您可以用列表理解替换 map/tuple 对，它可以是异步的，并且很容易转换为元组:

some_tuple = tuple([await fn(elem) for elem in iterable])

另一种选择是使用 asyncstdlib 中的 map 和 tuple 的异步版本包，它还提供了许多额外的有用功能:

import asyncstdlib.builtins.map as amap
import asyncstdlib.builtins.tuple as atuple

some_tuple = await atuple(amap(fn, iterable))