Python多進程與單進程效率對比


運行環境:Python3 in win10
先生成200個測試文件

# generate.py
i = 0
while i < 200:
    o = open("test\\" + str(i) + ".py", "w")
    content = str(i)
    o.write(content)
    o.close()
    i += 1

多進程拷貝文件

# multi-pool-copy.py
from multiprocessing import Pool, Manager
import time
import os
import shutil
import random

start = time.time()

def copyFile(file_name, old_folder_name, new_folder_name, q):
    time.sleep(random.random())
    shutil.copyfile(old_folder_name + '\\' + file_name, new_folder_name + '\\' + file_name,)
    q.put(file_name)  # put item into the queue

def main():
    pool = Pool(5)
    q = Manager().Queue()

    old_folder_name = input("Please input the folder name you want to copy: ")

    new_folder_name = old_folder_name + "-copy"
    os.mkdir(new_folder_name)

    file_name_list = os.listdir(old_folder_name)

    for file in file_name_list:
        pool.apply_async(copyFile, args=(file, old_folder_name, new_folder_name, q))
    
    cnt = 0
    allLength = len(file_name_list)
    while cnt < allLength:
        message = q.get()
        cnt += 1
        print("\rCopying %s, Process Bar is:%d%%" % (message, (cnt / allLength) * 100), end="")
    print("Copy Done!")

if __name__ == "__main__":
    main()
    end = time.time()
    print("Time-consuming: %#.2fs" % (end-start))

在使用單進程拷貝文件之前,需要手動刪除test-copy文件夾

# single-pool-copy.py
import time 
import os
import shutil
import random

start = time.time()

def copyFile(file_name, old_folder_name, new_folder_name):
    time.sleep(random.random())
    shutil.copyfile(old_folder_name + '\\' + file_name, new_folder_name + '\\' + file_name, )


def main():
    old_folder_name = input("Please input the folder name you want to copy: ")

    new_folder_name = old_folder_name + "-copy"
    os.mkdir(new_folder_name)

    file_name_list = os.listdir(old_folder_name)

    cnt = 0
    allLength = len(file_name_list)

    for file in file_name_list:
        copyFile(file, old_folder_name, new_folder_name)
        cnt += 1
        print("\rCopying %s, Process Bar is:%d%%" % (file, (cnt / allLength) * 100), end="")        
    print("Copy Done!")

if __name__ == "__main__":
    main()
    end = time.time()
    print("Time-consuming: %#.2fs" % (end-start))


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM