Skip to content

Instantly share code, notes, and snippets.

@vikaslalwani
Forked from baojie/hello_multiprocessing.py
Last active May 5, 2021 12:11
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save vikaslalwani/8461d705433625621dec1aea734df915 to your computer and use it in GitHub Desktop.
Save vikaslalwani/8461d705433625621dec1aea734df915 to your computer and use it in GitHub Desktop.
Python multiprocessing hello world. Split a list and process sublists in different jobs
import multiprocessing
# split a list into evenly sized chunks
def chunks(l, n):
return [l[i:i+n] for i in range(0, len(l), n)]
def do_job(job_id, data_slice):
for item in data_slice:
print "job", job_id, item
def dispatch_jobs(data, job_number):
total = len(data)
chunk_size = total / job_number
slice = chunks(data, chunk_size) # for py3.7+ -> slice = chunks(data, int(chunk_size)) OR chunk_size = total // job_number
jobs = []
for i, s in enumerate(slice):
j = multiprocessing.Process(target=do_job, args=(i, s))
jobs.append(j)
for j in jobs:
j.start()
if __name__ == "__main__":
data = ['a', 'b', 'c', 'd']
dispatch_jobs(data, 2)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment