简化循环代码---向量化
计算1~5的平方和
循环代码如下:
A=[1,2,3,4,5]
B=[1,2,3,4,5]
res=0
for i in range(0,5):
res+=A[i]*B[i]
向量化代码如下:
import numpy as np
A=[1,2,3,4,5]
B=[1,2,3,4,5]
res=np.dot(A,B.T)
上面这段代码中最关键的一行是np.dot(A,B.T),dot方法执行了一个向量乘法运算。这样处理后据说能提高300倍的运行速度~~~~~
...
原创
2021-02-03 20:54:44 ·
279 阅读 ·
0 评论