在python中启动多个线程

时间:2013-12-11 12:24:33

标签: python multithreading

我希望下面的代码中的saveData函数能够同时执行。但相反,在第一次运行saveData之后我得到了错误:

Traceback (most recent call last):
  File "preProcess.py", line 70, in <module>
    run()        
  File "preProcess.py", line 61, in run
    thread.start_new_thread(saveData(slice1, slice2, slice3, dset), ("Thread-" + str(i), 1, ) )
TypeError: first arg must be callable

我的代码是

#!/usr/bin/env python
import sys

import numpy as np
import h5py
import scipy
from PIL import Image
import timeit
import thread

import matplotlib.pyplot as plt

def saveImage(array, filename):
  fig=plt.figure(figsize=(4,3))
  ax=fig.add_subplot(1,1,1)
  plt.axis('off')
  p = plt.imshow(array)
  p.set_cmap('gray')
  extent = ax.get_window_extent().transformed(fig.dpi_scale_trans.inverted())
  plt.savefig(filename, bbox_inches=extent) 

def saveData(value1, value2, value3, dset):
  filename = "tomo1_" + str(value1) + ".png" 
  data = dset[value1::]
  saveImage(data, filename)
  filename = "tomo2_" + str(value2) + ".png" 
  data = dset[:value2:]
  saveImage(data, filename)
  filename = "tomo3_" + str(value3) + ".png" 
  data = dset[::value3]
  saveImage(data, filename)

def run():

  # Reopen the file and dataset using default properties.
  f = h5py.File(sys.argv[1])
  dset = f[sys.argv[2]]

  dim1 = len(dset)
  dim2 = len(dset[0])
  dim3 = len(dset[0][0])

  slice1 = 0
  slice2 = 0
  slice3 = 0
  factor1 = dim1/48
  factor2 = dim2/48
  factor3 = dim3/48
  tic=timeit.default_timer()
  for i in range(0,48):
    thread.start_new_thread(saveData(slice1, slice2, slice3, dset), ("Thread-" + str(i),     1, ) )
    slice1 = slice1 + factor1
    slice2 = slice2 + factor2
    slice3 = slice3 + factor3

  toc=timeit.default_timer()
  print "elapsed time: " + str(toc - tic)

if __name__ == "__main__":
    run()        

我想同时执行48次saveData函数。 我做错了什么?

2 个答案:

答案 0 :(得分:8)

Gryphius指出:你必须执行saveData中的start_new_thread,而是将你的函数用作thread.start_new_thread的普通参数!

原因是,thread.start_new_thread将在Thread中为您执行该功能。所以你不仅要传递函数作为参数(不执行它),还要传递函数的参数:

thread.start_new_thread(saveData,(slice1,slice2,slice3,dset),("Thread-"+str(i),1,))

如您所见,函数的参数作为元组传递。

作为旁注: 这个成语,不是调用函数,而是将它与它的参数一起传递,也可以在事件处理函数中看到:回调。

另外考虑使用Threading包而不是thread,它更复杂,让您拥有更多的自由和功能!

答案 1 :(得分:5)

仔细阅读thread.start_new_thread的api文档。它期望第一个参数是可调用的,因此在您的情况下只需saveData而不是saveData(slice1, slice2, slice3, dset)。第二个参数是带有函数参数的元组。

尝试:

thread.start_new_thread(saveData,(slice1, slice2, slice3, dset))