我有一个python脚本,应该并行运行多个作业。我将最大进程设置为20但我需要脚本在发送作业之间休眠5秒。 所以这是我的示例代码:
#!/usr/bin/env python
import multiprocessing
import subprocess
def prcss(cmd):
sbc = subprocess.call
com = sbc(cmd, shell='True')
return (com)
if __name__=='__main__':
s = 'sleep 5'
cmd= []
for j in range(1,21):
for i in range(10):
sis = "nohup ~/mycodes/code > str(j)"+"/"+"out"+str(i)+".dat"
cmd.append(sis)
cmd.append(s)
pool=multiprocessing.Pool(processes=20)
pool.map(prcss,cmd)
虽然我在'sis'命令之间睡了5,但是当我运行我的脚本时,所有工作都会立即启动。我需要在'sis'命令之间休眠,因为每个作业的输出取决于计算机时钟。因此,如果我运行20个作业,它们都以相同的系统时钟开始,因此它们都将具有相同的输出。
知道如何让我的脚本在'sis'命令之间睡觉吗?
Abedin
答案 0 :(得分:2)
看看docs for pool.map()。创建项目列表然后使用map将它们提交到池时,所有作业将一起提交到池中。由于您有20个工作流程,因此您的20个工作将同时启动(有效)。这包括您的sis
命令和睡眠命令。甚至没有保证它们将以相同的顺序执行和完成,只是您将以相同的顺序收到结果。 apply_async()函数可能对您更好,因为您可以控制何时将作业提交到池中。
在我看来,你希望你的Python脚本在发出sis
命令之前等待5秒钟,所以你没有理由需要在工作进程中执行sleep命令。尝试重构成这样的东西:
import multiprocessing
import subprocess
import time
def prcss(cmd):
# renaming the subprocess call is silly - remove the rename
com = subprocess.call(cmd, shell='True')
return (com)
if __name__=='__main__':
pool = multiprocessing.Pool(processes=20)
results_objects = []
for j in range(1,21):
for i in range(10):
sis = 'nohup ~/mycodes/code >'+str(j)+'/'+'out'+str(i)+'.dat'
# make an asynchronous that will execute our target function with the
# sis command
results_objects.append(pool.apply_async(prcss, args=(sis,))
# don't forget the extra comma in the args - Process must receive a tuple
# now we pause for five sections before submitting the next job
time.sleep(5)
# close the pool and wait for everything to finish
pool.close()
pool.join()
# retrieve all of the results
result = [result.get() for result in results_objects]
另一个注意事项:由于语法高亮显示已应用,因此很容易看到您在sis
字符串中缺少结束引号,也可能是'+'。请考虑使用string.format():
sis = 'nohup ~/mycodes/code > {}/out{}.dat'.format(j, i)
如果反斜杠用于分隔路径层次结构,则应使用os.path.join():
import os
sis = os.path.join('nohup ~/mycodes/code > {}'.format(j), 'out{}.dat'.format(i))
生成的第一个字符串(在任何一种情况下)都是:
nohup~ / mycodes / code> 1 / out0.dat