简化循环代码---向量化

计算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)

上面这段代码中最关键的一行是np.dot(A,B.T),dot方法执行了一个向量乘法运算。这样处理后据说能提高300倍的运行速度~~~~~

猜你喜欢

转载自blog.csdn.net/qq_42697271/article/details/113617132