我正在尝试并行进行一些绘图以更快地完成大批量作业。为此,我为我计划制作的每个情节开始一个线程。
我曾希望每个线程都能完成其绘图并自行关闭(据我了解,Python 在线程完成 run() 中的所有语句时会关闭线程)。下面是一些显示此行为的代码。
如果创建图形的行被注释掉,它将按预期运行。另一个似乎很有帮助的花絮是,当您只生成一个线程时,它也会按预期运行。
import matplotlib.pyplot as plt
import time
import Queue
import threading
def TapHistplots():
## for item in ['str1']:
# # it behaves as expected if the line above is used instead of the one below
for item in ['str1','str2']:
otheritem = 1
TapHistQueue.put((item, otheritem))
makeTapHist().start()
class makeTapHist(threading.Thread):
def run(self):
item, otheritem = TapHistQueue.get()
fig = FigureQueue.get()
FigureQueue.put(fig+1)
print item+':'+str(fig)+'\n',
time.sleep(1.3)
plt.figure(fig) # comment out this line and it behaves as expected
plt.close(fig)
TapHistQueue = Queue.Queue(0)
FigureQueue = Queue.Queue(0)
def main():
start = time.time()
"""Code in here runs only when this module is run directly"""
FigureQueue.put(1)
TapHistplots()
while threading.activeCount()>1:
time.sleep(1)
print 'waiting on %d threads\n' % (threading.activeCount()-1),
print '%ds elapsed' % (time.time()-start)
if __name__ == '__main__':
main()
任何帮助都将受到应有的赞赏。