我希望下面的代码中的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函数。 我做错了什么?
答案 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))