利用Python进行数据分析(6) NumPy基础: 矢量计算

时间:2021-08-22 14:38:54
矢量化指的是用数组表达式代替循环来操作数组里的每个元素。
NumPy提供的通用函数(既ufunc函数)是一种对ndarray中的数据进行元素级别运算的函数。
例如,square函数计算各元素的平方,rint函数将各元素四舍五入:
利用Python进行数据分析(6) NumPy基础: 矢量计算
还有一些函数接受2个参数,叫二元ufunc,比如add函数和maximum函数:
利用Python进行数据分析(6) NumPy基础: 矢量计算

numpy.where函数

numpy.where函数是三元表达式 x if condition else y 的矢量化版本,例如:
利用Python进行数据分析(6) NumPy基础: 矢量计算
np.where函数的第二个参数和第三个参数不是必要的,它们都可以是标量值,例如:
利用Python进行数据分析(6) NumPy基础: 矢量计算

数学和统计方法

例如np.sum函数可以对数组里的元素求和:
利用Python进行数据分析(6) NumPy基础: 矢量计算
对于二维数组,sum函数也是将所有元素求和,但是二维数组是有横轴和竖轴两个方向的,所以sum函数对于二维数组还可以按照方向进行求和:
利用Python进行数据分析(6) NumPy基础: 矢量计算