python 多线程稀疏矩阵乘法

 import threading, time
 import numpy as np
 res = []
 class MyThread(threading.Thread):
     def __init__(self,i,j,m1,m2):
         threading.Thread.__init__(self)
         self.x, self.y = i,j
         self.m1, self.m2 = m1, m2
     def run(self):
         global res, lock
         if lock.acquire():
             m1 = self.m1[self.m1[:,0]==self.x]
             m2 = self.m2[self.m2[:,1]==self.y]
             value = 0.
             for item1 in m1:
                 for item2 in m2:
                     if item1[1] == item2[0]:
                         value += item1[2]*item2[2]
             res.append([self.x,self.y,value])
             lock.release()
 if "__main__" == __name__:
     m1 = [[2,2],[0,0,1],[0,1,2],[1,0,3],[1,1,4]]
     m2 = [[2,3],[0,0,2],[0,2,1],[1,2,3],[1,1,4]]
     s1, s2 = m1[0], m2[0]
     assert s1[1]==s2[0], 'mismatch'
     m1_value = np.array(m1[1:])
     m2_value = np.array(m2[1:])
     rows, cols = s1[0], s2[1]
     res.append([rows, cols])
     ThreadList = []
     lock = threading.Lock()
     for i in range(rows):
         for j in range(cols):
             t = MyThread(i,j,m1_value,m2_value)
             ThreadList.append(t)
     for t in ThreadList:
         t.start()
     for t in ThreadList:
         t.join()
     print (res)