Python入门教程(三):史上最全的Numpy计算函数总结,建议收藏!

2020-12-30 14:23:31 浏览数 (1)

Numpy提供了灵活的、静态类型的、可编译的程序接口口来优化数组的计算,也被称作向量操作,因此在Python数据科学界Numpy显得尤为重要。Numpy的向量操作是通过通用函数实现的。今天小编会给大家较为全面地介绍下Numpy的通用函数。

01

数组的运算

Numpy通用函数涉及到Python原生的算术运算符,标准的加减乘除都可以使用,同时这些运算符也是Numpy内置函数的简单封装器,例如“ ”就是add函数的封装器。下图汇总了Numpy实现的算术运算符。

Numpy的加减乘除运算

代码语言:javascript复制
x = np.arange(4)
print("x     =", x)
print("x   5 =", x   5)
print("x - 5 =", x - 5)
print("x * 2 =", x * 2)
print("x / 2 =", x / 2)
print("x // 2 =", x // 2)  # 向下整除

# x     = [0 1 2 3]
# x   5 = [5 6 7 8]
# x - 5 = [-5 -4 -3 -2]
# x * 2 = [0 2 4 6]
# x / 2 = [ 0.   0.5  1.   1.5]
# x // 2 = [0 0 1 1]

求负数、**表示求指数运算符以及%表示求%运算符,请看下面示例:

代码语言:javascript复制
print("x ** 2 = ", x ** 2)
print("x % 2  = ", x % 2)
# -x     =  [ 0 -1 -2 -3]
# x ** 2 =  [0 1 4 9]
# x % 2  =  [0 1 0 1]

如果多个运算符组合使用时,需要考虑这些运算符的一个优先级。

代码语言:javascript复制
-(0.5*x   1) ** 2
# array([-1.  , -2.25, -4.  , -6.25])

使用封装函数计算标准的加减乘除。

代码语言:javascript复制
np.add(x, 2)
# array([2, 3, 4, 5])

02

绝对值

Numpy通用的绝对值函数是np.absolute,也可以用其别名来访问np.abs。这个通用函数也可以处理复数,处理复数时,绝对值返回的是该复数的模。

代码语言:javascript复制
x = np.array([-2, -1, 0, 1, 2])
abs(x)
# array([2, 1, 0, 1, 2])

np.absolute(x)
#  array([2, 1, 0, 1, 2])
np.abs(x)
#  array([2, 1, 0, 1, 2])

x = np.array([3 - 4j, 4 - 3j, 2   0j, 0   1j])
np.abs(x)
# array([ 5.,  5.,  2.,  1.])
代码语言:javascript复制

03

三角函数

三角函数是数据科学中常用到的函数,这里会讲解三角函数的计算以及逆三角函数的计算。

首先,我们需要先定义一个角度数组,然后通过cos(),sin(),tan()等三角函数进行计算。

代码语言:javascript复制
theta = np.linspace(0, np.pi, 3)

print("theta      = ", theta)
print("sin(theta) = ", np.sin(theta))
print("cos(theta) = ", np.cos(theta))
print("tan(theta) = ", np.tan(theta))

# theta      =  [ 0.          1.57079633  3.14159265]
# sin(theta) =  [  0.00000000e 00   1.00000000e 00   1.22464680e-16]
# cos(theta) =  [  1.00000000e 00   6.12323400e-17  -1.00000000e 00]
# tan(theta) =  [  0.00000000e 00   1.63312394e 16  -1.22464680e-16]

同样我们也可以计算逆三角函数。

代码语言:javascript复制
x = [-1, 0, 1]
print("x         = ", x)
print("arcsin(x) = ", np.arcsin(x))
print("arccos(x) = ", np.arccos(x))
print("arctan(x) = ", np.arctan(x))

# x         =  [-1, 0, 1]
# arcsin(x) =  [-1.57079633  0.          1.57079633]
# arccos(x) =  [ 3.14159265  1.57079633  0.        ]
# arctan(x) =  [-0.78539816  0.          0.78539816]

04

指数和对数

Numpy中的指数运算。

代码语言:javascript复制
x = [1, 2, 3]
print("x     =", x)
print("e^x   =", np.exp(x))
print("2^x   =", np.exp2(x))
print("3^x   =", np.power(3, x))

# x     = [1, 2, 3]
# e^x   = [  2.71828183   7.3890561   20.08553692]
# 2^x   = [ 2.  4.  8.]
# 3^x   = [ 3  9 27

Numpy中的对数运算。

对数运算是指数运算的逆运算,最基础的np.log是以自然对数为底数的对数,同时也可以使用np.log2,np.log10等计算以2或10为底的对数。

代码语言:javascript复制
x = [1, 2, 4, 10]
print("x        =", x)
print("ln(x)    =", np.log(x))
print("log2(x)  =", np.log2(x))
print("log10(x) =", np.log10(x))

# x        = [1, 2, 4, 10]
# ln(x)    = [ 0.          0.69314718  1.38629436  2.30258509]
# log2(x)  = [ 0.          1.          2.          3.32192809]
# log10(x) = [ 0.          0.30103     0.60205999  1.        ]

特殊情况下,对于非常小的输入值可以保持较好的精度。当x很小时,以下函数给出的值np.log和np.exp的计算更加精确。

代码语言:javascript复制
x = [0, 0.001, 0.01, 0.1]
print("exp(x) - 1 =", np.expm1(x))
print("log(1   x) =", np.log1p(x))

# exp(x) - 1 = [ 0.          0.0010005   0.01005017  0.10517092]
# log(1   x) = [ 0.          0.0009995   0.00995033  0.09531018]

05

专用的通用函数

Numpy还提供了很多通用函数,包括了双曲三角函数,比特位运算,比较运算符,弧度转化为角度的运算,取整和求余运算。除此之外呢,Python中还有更加专用的通用函数模块scipy.special,下面会为大家展示一部分的代码片段。

Gamma函数(广义阶乘,generlized factorials)和相关函数

代码语言:javascript复制
from scipy import special
x = [1, 5, 10]
print("gamma(x)     =", special.gamma(x))
print("ln|gamma(x)| =", special.gammaln(x))
print("beta(x, 2)   =", special.beta(x, 2))

# gamma(x)     = [  1.00000000e 00   2.40000000e 01   3.62880000e 05]
# ln|gamma(x)| = [  0.           3.17805383  12.80182748]
# beta(x, 2)   = [ 0.5         0.03333333  0.00909091]

误差函数(高斯积分)

高斯积分的实现和逆实现

代码语言:javascript复制
# Error function (integral of Gaussian)
# its complement, and its inverse
x = np.array([0, 0.3, 0.7, 1.0])
print("erf(x)  =", special.erf(x))
print("erfc(x) =", special.erfc(x))
print("erfinv(x) =", special.erfinv(x))

# erf(x)  = [ 0.          0.32862676  0.67780119  0.84270079]
# erfc(x) = [ 1.          0.67137324  0.32219881  0.15729921]
# erfinv(x) = [ 0.          0.27246271  0.73286908         inf]

06

指定输出

所有的通用函数都可以通过out参数来指定计算结果的存放位置。

代码语言:javascript复制
x = np.arange(5)
y = np.empty(5)
np.multiply(x, 10, out=y)
print(y)
# [  0.  10.  20.  30.  40.]
代码语言:javascript复制
这个特性也可以被称为数组视图,例如将计算结果写入指定数组的每隔一个元素的位置。
代码语言:javascript复制
y = np.zeros(10)
np.power(2, x, out=y[::2])
print(y)
# [  1.   0.   2.   0.   4.   0.   8.   0.  16.   0.]

07

聚合

我们希望用一个特定的运算reduce一个数组,那么可以用任何通用函数的reduce方法。一个reduce方法会对给定元素和操作重复执行,直到得到这个结果。

对add通用函数调用reduce方法会返回数组中所有元素的和。

代码语言:javascript复制
x = np.arange(1, 6)
np.add.reduce(x)
# 15

对multiply通用函数调用reduce方法会返回数组中所有元素的乘积。

代码语言:javascript复制
np.multiply.reduce(x)
# 120

accumulate函数可以储存每次计算的中间结果表。

代码语言:javascript复制
np.add.accumulate(x)
# array([ 1,  3,  6, 10, 15]
代码语言:javascript复制
np.multiply.accumulate(x)
array([  1,   2,   6,  24, 120])

# array([  1,   2,   6,  24, 120])

任何通用函数都可以用outer方法获得两个不同输入数组所有元素对函数运算的结果。这意味着一行代码实现一个乘法表。

通用函数还能够操纵形状和大小不一样的数组,一组这样的操作被称为广播,后面会细讲。

代码语言:javascript复制
x = np.arange(1, 6)
np.multiply.outer(x, x)
# array([[ 1,  2,  3,  4,  5],
#        [ 2,  4,  6,  8, 10],
#        [ 3,  6,  9, 12, 15],
#        [ 4,  8, 12, 16, 20],
#        [ 5, 10, 15, 20, 25]])

全部代码已上传,公众号后台回复【Numpy计算函数】即可获得。

0 人点赞