我正在使用Python的concurrent.futures框架。我使用了map()函数来启动并发任务:
def func(i):
return i*i
list = [1,2,3,4,5]
async_executor = concurrent.futures.ThreadPoolExecutor(5)
results = async_executor.map(func,list)我只对第一个n结果感兴趣,并且希望在第一个n线程完成后停止执行程序,其中n小于输入列表的大小。在Python中有什么方法可以做到吗?我还需要研究另一个框架吗?
发布于 2018-06-02 02:04:38
您不能为此使用map(),因为它提供了无法停止等待结果的方法,也无法获得提交的期货并取消它们。但是,您可以使用submit()来完成
import concurrent.futures
import time
def func(i):
time.sleep(i)
return i*i
list = [1,2,3,6,6,6,90,100]
async_executor = concurrent.futures.ThreadPoolExecutor(2)
futures = {async_executor.submit(func, i): i for i in list}
for ii, future in enumerate(concurrent.futures.as_completed(futures)):
print(ii, "result is", future.result())
if ii == 2:
async_executor.shutdown(wait=False)
for victim in futures:
victim.cancel()
break上面的代码大约需要11秒才能运行--它执行作业1,2,3,6,7,而不执行其余的作业。
https://stackoverflow.com/questions/50652724
复制相似问题