这应该是我的第三个也是最后一个问题,关于我尝试提高我使用 python 进行的一些统计分析的性能。我的代码有 2 个版本(单核与多处理),我希望通过使用多核来获得性能,因为我希望我的代码可以解压缩/解压缩很多二进制字符串,遗憾的是我注意到性能实际上是通过使用多个核心。
我想知道是否有人对我观察到的情况有可能的解释(向下滚动到 4 月 16 日更新以获取更多信息)?
程序的关键部分是函数 numpy_array (+ 多处理中的解码),下面的代码片段(可通过 pastebin 访问的完整代码,进一步如下):
def numpy_array(data, peaks):
rt_counter=0
for x in peaks:
if rt_counter %(len(peaks)/20) == 0:
update_progress()
peak_counter=0
data_buff=base64.b64decode(x)
buff_size=len(data_buff)/4
unpack_format=">%dL" % buff_size
index=0
for y in struct.unpack(unpack_format,data_buff):
buff1=struct.pack("I",y)
buff2=struct.unpack("f",buff1)[0]
if (index % 2 == 0):
data[rt_counter][1][peak_counter][0]=float(buff2)
else:
data[rt_counter][1][peak_counter][1]=float(buff2)
peak_counter+=1
index+=1
rt_counter+=1
多处理版本通过一组函数执行此操作,我将在下面显示关键 2:
def tonumpyarray(mp_arr):
return np.frombuffer(mp_arr.get_obj())
def numpy_array(shared_arr,peaks):
processors=mp.cpu_count()
with contextlib.closing(mp.Pool(processes=processors,
initializer=pool_init,
initargs=(shared_arr, ))) as pool:
chunk_size=int(len(peaks)/processors)
map_parameters=[]
for i in range(processors):
counter = i*chunk_size
chunk=peaks[i*chunk_size:(i+1)*chunk_size]
map_parameters.append((chunk, counter))
pool.map(decode,map_parameters)
def decode ((chunk, counter)):
data=tonumpyarray(shared_arr).view(
[('f0','<f4'), ('f1','<f4',(250000,2))])
for x in chunk:
peak_counter=0
data_buff=base64.b64decode(x)
buff_size=len(data_buff)/4
unpack_format=">%dL" % buff_size
index=0
for y in struct.unpack(unpack_format,data_buff):
buff1=struct.pack("I",y)
buff2=struct.unpack("f",buff1)[0]
#with shared_arr.get_lock():
if (index % 2 == 0):
data[counter][1][peak_counter][0]=float(buff2)
else:
data[counter][1][peak_counter][1]=float(buff2)
peak_counter+=1
index+=1
counter+=1
可以通过这些 pastebin 链接访问完整的程序代码
我使用包含 239 个时间点和每个时间点约 180k 测量对的文件观察到的性能对于单核约为 2.5m,对于多处理约为 3.5。
PS:前两个问题(我第一次尝试并行化):
——4月16日——
我一直在使用 cProfile 库对我的程序进行分析(cProfile.run('main()')
在 中__main__
,这表明有一个步骤会减慢一切:
ncalls tottime percall cumtime percall filename:lineno(function)
23 85.859 3.733 85.859 3.733 {method 'acquire' of 'thread.lock' objects}
我在这里不明白的是thread.lock
对象用于threading
(据我了解)但不应该用于多处理,因为每个核心都应该运行一个线程(除了拥有自己的锁定机制),那么这是怎么发生的为什么一次通话需要 3.7 秒?