簡體   English   中英

Python多處理:將簡單的工作拆分為多個進程

[英]Python multiprocessing: Simple job split across many processes

編輯

提議的代碼實際上有效! 我只是在沒有顯示輸出的IDE中運行它。

我將問題拋在腦后,因為評論/答案很有啟發性


我需要將大量工作分配給許多工人。 為了弄清楚如何做到這一點,我使用了以下簡單示例,其中的代碼大部分來自此處 基本上,我正在獲取一個列表,將其分解成較短的子列表(塊),並要求multiprocessing使用專門的工作人員打印每個子列表的內容:

import multiprocessing
from math import ceil

# Breaking up the long list in chunks:
def chunks(l, n):
    return [l[i:i+n] for i in range(0, len(l), n)]

# Some simple function 
  def do_job(job_id, data_slice):
      for item in data_slice:
          print("{}_{}".format(job_id, item))

然后,我這樣做:

if __name__ == '__main__':

    # My "long" list
    l = [letter for letter in 'abcdefghijklmnopqrstuvwxyz']

    my_chunks = chunks(l, ceil(len(l)/4))

此時,my_chunks符合預期:

[['a', 'b', 'c', 'd', 'e', 'f', 'g'],
 ['h', 'i', 'j', 'k', 'l', 'm', 'n'],
 ['o', 'p', 'q', 'r', 's', 't', 'u'],
 ['v', 'w', 'x', 'y', 'z']]

然后:

    jobs = []
    for i, s in enumerate(my_chunks):
        j = mp.Process(target=do_job, args=(i, s))
        jobs.append(j)
    for j in jobs:
        print('starting job {}'.format(str(j)))        
        j.start()

最初 ,我寫這個問題是因為我沒有從do_job函數中獲得預期的打印輸出。

事實證明 ,從命令行運行時,代碼工作正常。

也許這是您第一次進行多處理? 您是等待流程退出還是要退出主要流程,然后流程才有時間完成那里的工作?

from multiprocessing import Process
from string import ascii_letters
from time import sleep


def job(chunk):
    done = chunk[::-1]
    print(done)

def chunk(data, parts):
    divided = [None]*parts
    n = len(data) // parts
    for i in range(parts):
        divided[i] = data[i*n:n*(i+1)]
    if len(data) % 2 != 0:
        divided[-1] += [data[-1]]
    return divided


def main():
    data = list(ascii_letters)
    workers = 4
    data_chunks = chunk(data, workers)
    ps = []
    for i in range(4):
        w = Process(target=job, args=(data_chunks[i],))
        w.deamon = True
        w.start()
        ps += [w]
    sleep(2)



if __name__ == '__main__':
    main()

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM