一、向量化的概念和好处
- 向量化:使用NumPy数组的数组表达式完成多种数据操作任务,而无须写些大量循环。
这种利用数组表达式来替代显式循环的方法,称为向量化 - 向量化的好处:加快运行速度,向量化的数组操作会比纯Python的等价实现在速度上快一到两个数量级(甚至更多)
二、一个例子
对一些二维数据来计算函数sqrt(x^2 + y^2)的值。
- 使用np.meshgrid函数接收两个一维数组,并根据两个数组的所有(x,y)对生成一个二维矩阵:
import numpy as np
import matplotlib.pyplot as plt
# 在-5到5的范围内,以0.01为间隔生成一个数组
points = np.arange(-5, 5, 0.01) # 1000 equally spaced points
print("**************")
print(points)
xs, ys = np.meshgrid(points, points)
print("**************")
print(xs)
print("-------------")
print(ys)
结果
**************
[-5.0000000e+00 -4.9900000e+00 -4.9800000e+00 -4.9700000e+00
-4.9600000e+00 -4.9500000e+00 -4.9400000e+00 -