欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Python中运行并行任务技巧

程序员文章站 2023-08-12 13:44:46
示例 标准线程多进程,生产者/消费者示例: worker越多,问题越大 复制代码 代码如下: # -*- coding: utf8 -*- import os...

示例

标准线程多进程,生产者/消费者示例:
worker越多,问题越大

复制代码 代码如下:

# -*- coding: utf8 -*-

import os
import time
import queue
import threading
from pil import image

def create_thumbnail(filename, size=(128, 128)):
    try:
        fp, fmt = filename.rsplit('.', 1)
        im = image.open(filename)
        im.thumbnail(size, image.antialias)
        im.save((fp + '_'+'x'.join(str(i) for i in size) + '.'+fmt), im.format)
        return '%s thumbnail success!' % filename
    except exception:
        return '%s thumbnail failed!' % filename


def get_image_paths(folder):
    return [os.path.join(folder, f) for f in os.listdir(folder) if 'png' in f]


class consumer(threading.thread):
    def __init__(self, queue):
        threading.thread.__init__(self)
        self._queue = queue

    def run(self):
        while true:
            content = self._queue.get()
            if isinstance(content, str) and content == 'quit':
                break
            respone = create_thumbnail(content)
        print 'bye bye!'


def producer():
    filenames = get_image_paths('images')
    queue = queue.queue()
    worker_threads = build_worker_pool(queue, 4)
    start_time = time.time()

    for filename in filenames:
        queue.put(filename)
    for worker in worker_threads:
        queue.put('quit')
    for worker in worker_threads:
        worker.join()

    print time.time() - start_time


def build_worker_pool(queue, size):
    workers = []
    for _ in range(size):
        worker = consumer(queue)
        worker.start()
        workers.append(worker)
    return workers


if __name__ == '__main__':
    producer()

map

map能够处理集合按顺序遍历,最终将调用产生的结果保存在一个简单的集合当中。

Python中运行并行任务技巧

复制代码 代码如下:

# -*- coding: utf8 -*-

import os
import time
from multiprocessing import pool
from pil import image

def create_thumbnail(filename, size=(128, 128)):
    try:
        fp, fmt = filename.rsplit('.', 1)
        im = image.open(filename)
        im.thumbnail(size, image.antialias)
        im.save((fp + '_'+'x'.join(str(i) for i in size) + '.'+fmt), im.format)
        return '%s thumbnail success!' % filename
    except exception:
        return '%s thumbnail failed!' % filename


def get_image_paths(folder):
    return [os.path.join(folder, f) for f in os.listdir(folder) if 'png' in f]


def main():
    filenames = get_image_paths('images')
    start_time = time.time()
   
    pool = pool(4)
    pool.map(create_thumbnail, filenames)
    pool.close()
    pool.join()

    print time.time() - start_time


if __name__ == '__main__':
    main()