Python数据处理(3)-NumPy的矢量化计算

Numpy数组使你可以用数组表达式代替循环,这种做法通常被称为矢量化,这种计算方法更加简洁,例如深度学习框架TensorFlow中的基本操作就是基于张量数组的。

前面我们已经介绍了基本的数组算数运算,接下来将介绍另外两种常用的矢量化操作。

1.meshgrid

np.meshgrid函数可以接受两个一维数组,并产生两个二维数组,生成的二维数组可以用于绘制三维图像。以绘制z = x^2 + y^2三维图像为例:

(1)首先生成-3到3的等差数列数组,数组大小为101,

(2)使用np.meshgrid函数生成二维数组x和y,数组大小为101*101

(3)数组算数运算得到z

(4)画图

2.where

np.where函数是三元表达式x if condition else y的矢量化版本,便于理解,我们给出两个例子:

(1)x if x > y else y

(2)0 if x < 0.5 else 1

  • 发表于:
  • 原文链接http://kuaibao.qq.com/s/20171222G0DTMC00?refer=cp_1026
  • 腾讯「腾讯云开发者社区」是腾讯内容开放平台帐号(企鹅号)传播渠道之一,根据《腾讯内容开放平台服务协议》转载发布内容。
  • 如有侵权,请联系 cloudcommunity@tencent.com 删除。

扫码关注腾讯云开发者

领取腾讯云代金券