Python多进程问题：如何获取首个匹配结果并终止进程

你的问题核心在于**pool.map()是阻塞式的，它会等待所有子进程完成任务后才返回全部结果**，所以你看到的输出其实是所有任务都执行完毕后的结果，完全没法实现“找到第一个匹配就立即返回”的并行效果。而且你循环里的break逻辑也不对——此时所有任务都已经跑完了，break只是停止打印而已。

修复方案：使用imap_unordered()实现异步结果迭代 #

我们可以用pool.imap_unordered()替代map()，它会按照子任务完成的先后顺序返回结果，而不是提交任务的顺序。一旦拿到第一个非None的匹配结果，就立即终止所有子进程，停止剩余任务的执行。

同时，我们可以简化代码，去掉不必要的sources字典和foo函数，直接把拆分好的片段作为参数传给子进程，逻辑更清晰。

修改后的完整代码：

import multiprocessing as mul

big = [[0,1],[3,0],[1,5],[0,3],[3,1],[0,6],[0,1],[3,2],[1,5],[0,3],[3,3],[0,6],[0,1],[3,4],[1,5],[0,3],[3,5],[0,6],[0,1],[3,6],[1,5],[0,3],[3,7],[0,6],[0,1],[3,8],[1,5],[0,3],[3,9],[0,6],[0,1],[3,10],[1,5],[0,3],[3,11],[0,6]]

def doit(segment):
    for item in segment:
        if item[0] == 3:
            return item
    return None

if __name__ == '__main__':
    # 直接拆分大列表为两元素片段的列表，无需字典存储
    segments = [big[2*x:2*x+2] for x in range(len(big)//2)]
    
    # 使用with语句自动管理进程池，无需手动close/join
    with mul.Pool(4) as pool:
        # imap_unordered按任务完成顺序返回结果
        for result in pool.imap_unordered(doit, segments):
            if result is not None:
                print(f"最快找到的匹配结果: {result}")
                # 找到结果后立即终止所有进程，停止剩余任务
                pool.terminate()
                break
        else:
            # 循环正常结束（没break），说明所有任务都没找到匹配
            print("未找到任何首元素为3的子列表")

关键修改点说明 #

• 替换map()为imap_unordered()：
  map()是阻塞的，必须等所有任务完成才返回结果列表；而imap_unordered()是异步迭代器，子进程完成一个任务就返回一个结果，顺序取决于任务完成速度。
• 立即终止进程池：
  一旦找到第一个匹配结果，调用pool.terminate()可以立即停止所有子进程，避免不必要的计算，真正实现“最快返回”的需求。
• 简化参数传递：
  直接生成segments列表并作为任务参数传入，避免了子进程访问全局字典的潜在问题（虽然这里是只读的，但传参更符合进程间通信的最佳实践）。
• 使用with语句管理进程池：
  自动处理进程池的关闭和资源释放，比手动调用close()和join()更安全简洁。

效果验证 #

运行修改后的代码，你会发现程序会在第一个完成匹配的子进程返回结果后立即终止，只会输出最快找到的那个匹配项，不会等待所有任务完成，完美体现并行查找的优势。

内容的提问来源于stack exchange，提问作者Tomasz Przemski