python经常遇到对一个list中的所有元素进行运算,比如求[1,2,3,4,5,6]中所有元素的正弦值,最简单的方式就是如下
import math
li=[]
for each in [1,2,3,4,5,6]:
li.append(math.sin(each*math.pi/180))
优化对循环的使用(即不使用循环),在numpy中可以实现像matlab中对数组的运算np
import numpy as np
te=np.array([1,2,3,4,5,6])
math_sin_vec = np.vectorize(math.sin)#sin函数不接受array传入,因此做函数向量化
sin_arr=math_sin_vec(te)
下面这种也可以
import numpy as np
te=np.array([1,2,3,4,5,6])
math_asin_vec = np.frompyfunc(math.asin, 1, 1)
sin_arr=math_sin_vec(te)