python在做数据处理方面有很大的优势,但是python的运行速度也一直是为诟病的。近期在工作中就遇到了大维度的矩阵乘法的运算,在作为服务时,响应时间略高,工程上需要在不部署其他分布式的前提下,优化运算速度。
接下来用不同的方式计算一个 100002400维 和 240010000维的矩阵并分别输出计算花费的时间。
numpy 方式
import numpy
aa = np.random.rand(10000, 2400)
bb = np.random.rand(2400, 10000)
%timeit cc = np.dot(aa, bb)
Output: 1 loop, best of 3: 9.21 s