数据分析第五讲：numpy的应用入门（四）

NumPy的应用（四）

向量

向量（vector）也叫矢量，是一个同时具有大小和方向，且满足平行四边形法则的几何对象。与向量相对的概念叫标量或数量，标量只有大小，绝大多数情况下没有方向。我们通常用带箭头的线段来表示向量，在平面直角坐标系中的向量如下图所示。需要注意的是，向量是表达大小和方向的量，并没有规定起点和终点，所以相同的向量可以画在任意位置，例如下图中$\mathbf{w}$和$\mathbf{v}$两个向量并没有什么区别。

向量有很多种代数表示法，对于二维空间的向量，下面几种写法都是可以的。
$$\mathbf{a}=⟨a_1,a_2⟩=(a_1,a_2)=\left(\begin{array}{c}{a}_1\\ a_2\end{array}\right)=\left[\begin{array}{c}{a}_1\\ a_2\end{array}\right]$$
向量的大小称为向量的模，它是一个标量，对于二维空间的向量，模可以通过下面的公式计算。
$$|\mathbf{a}|=\sqrt{a_1^2+a_2^2}$$
注意，这里的$|\mathbf{a}|$并不是绝对值，你可以将其称为向量$\mathbf{a}$的二范数，这是数学中的符号重用现象。上面的写法和概念也可以推广到$n$维空间，我们通常用${\mathbf{R}}^{\mathbf{n}}$表示$n$维空间，我们刚才说的二维空间可以记为${\mathbf{R}}^2$，三维空间可以记为${\mathbf{R}}^3$。虽然生活在三维空间的我们很难想象四维空间、五维空间是什么样子，但是这并不影响我们探讨高维空间，机器学习中，我们经常把有$n$个特征的训练样本称为一个$n$维向量。

向量的加法

相同维度的向量可以相加得到一个新的向量，运算的方法是将向量的每个分量相加，如下所示。
$$\mathbf{u}=\left[\begin{array}{c}{u}_1\\ u_2\\ ⋮\\ u_n\end{array}\right],\mathbf{v}=\left[\begin{array}{c}{v}_1\\ v_2\\ ⋮\\ v_n\end{array}\right],\mathbf{u}+\mathbf{v}=\left[\begin{array}{c}{u}_1+v_1\\ u_2+v_2\\ ⋮\\ u_n+v_n\end{array}\right]$$
向量的加法满足“平行四边形法则”，即两个向量$\mathbf{u}$和$\mathbf{v}$构成了平行四边形的两条邻边，相加的结果是平行四边形的对角线，如下图所示。

向量的数乘

一个向量$\mathbf{v}$可以和一个标量$k$相乘，运算的方法是将向量中的每个分量与该标量相乘即可，如下所示。
$$\mathbf{v}=\left[\begin{array}{c}{v}_1\\ v_2\\ ⋮\\ v_n\end{array}\right],k\cdot \mathbf{v}=\left[\begin{array}{c}k\cdot v_1\\ k\cdot v_2\\ ⋮\\ k\cdot v_n\end{array}\right]$$
我们可以用 NumPy 的数组来表示向量，向量的加法可以通过两个数组的加法来实现，向量的数乘可以通过数组和标量的乘法来实现，此处不再进行赘述。

向量的点积

点积（dot product）是两个向量之间最为重要的运算之一，运算的方法是将两个向量对应分量的乘积求和，所以点积的结果是一个标量，其几何意义是两个向量的模乘以二者夹角的余弦如下所示。
$$\begin{gathered} \mathbf{u}=\left[\begin{array}{c}{u}_1\\ u_2\\ ⋮\\ u_n\end{array}\right],\mathbf{v}=\left[\begin{array}{c}{v}_1\\ v_2\\ ⋮\\ v_n\end{array}\right] \\ \mathbf{u}\cdot \mathbf{v}=\sum _{i=1}^n{u}_i{v}_i=|\mathbf{u}||\mathbf{v}|cos\theta \end{gathered}$$
假如我们用3维向量来表示用户对喜剧片、言情片和动作片这三类电影的偏好，我们用1到5的数字来表示喜欢的程度，其中5表示非常喜欢，4表示比较喜欢，3表示无感，2表示比较反感，1表示特别反感。那么，下面的向量表示用户非常喜欢喜剧片，特别反感言情片，对动作片不喜欢也不反感。
$$\mathbf{u}=\left(\begin{array}{c}5\\ 1\\ 3\end{array}\right)$$
现在有两部电影上映了，一部属于言情喜剧片，一部属于喜剧动作片，我们把两部电影也通过3维向量的方式进行表示，如下所示。
$${\mathbf{m}}_1=\left(\begin{array}{c}4\\ 5\\ 1\end{array}\right),{\mathbf{m}}_2=\left(\begin{array}{c}5\\ 1\\ 5\end{array}\right)$$
如果现在我们需要向刚才的用户推荐一部电影，我们应该给他推荐哪一部呢？我们可以将代表用户的向量$\mathbf{u}$和代表电影的向量${\mathbf{m}}_1$和${\mathbf{m}}_2$分别进行点积运算，再除以向量的模长，得到向量夹角的余弦值，余弦值越接近1，说明向量的夹角越接近0度，也就是两个向量的相似度越高。很显然，我们应该向用户推荐跟他观影喜好相似度更高的电影。
$$\begin{gathered} cos{\theta }_1=\frac{\mathbf{u}\cdot \mathbf{m}1}{|\mathbf{u}||\mathbf{m}1|}\approx \frac{4\times 5+5\times 1+3\times 1}{5.92\times 6.48}\approx 0.73 \\ cos{\theta }_2=\frac{\mathbf{u}\cdot \mathbf{m}2}{|\mathbf{u}||\mathbf{m}2|}\approx \frac{5\times 5+1\times 1+3\times 5}{5.92\times 7.14}\approx 0.97 \end{gathered}$$
大家可能会说，向量${\mathbf{m}}_2$代表的电影肉眼可见跟用户是更加匹配的。的确，对于一个三维向量我们凭借直觉也能够给出正确的答案，但是对于一个$n$维向量，当$n$的值非常大时，你还有信心凭借肉眼的观察和本能的直觉给出准确的答案吗？向量的点积可以通过dot函数来计算，而向量的模长可以通过 NumPy 的linalg模块中的norm函数来计算，代码如下所示。

u = np.array([5, 1, 3])
m1 = np.array([4, 5, 1])
m2 = np.array([5, 1, 5])
print(np.dot(u, m1) / (np.linalg.norm(u) * np.linalg.norm(m1)))  # 0.7302967433402214
print(np.dot(u, m2) / (np.linalg.norm(u) * np.linalg.norm(m2)))  # 0.9704311900788593

向量的叉积

在二维空间，两个向量的叉积是这样定义的：
$$\begin{gathered} \mathbf{A}=\left(\begin{array}{c}{a}_1\\ a_2\end{array}\right),\mathbf{B}=\left(\begin{array}{c}{b}_1\\ b_2\end{array}\right) \\ \mathbf{A}\times \mathbf{B}=\left|\begin{array}{c}{a}_1{a}_2\\ b_1{b}_2\end{array}\right|=a_1{b}_2-a_2{b}_1 \end{gathered}$$
对于三维空间，两个向量的叉积结果是一个向量，如下所示：
$$\begin{gathered} \mathbf{A}=\left(\begin{array}{c}{a}_1\\ a_2\\ a_3\end{array}\right),\mathbf{B}=\left(\begin{array}{c}{b}_1\\ b_2\\ b_3\end{array}\right) \\ \mathbf{A}\times \mathbf{B}=\left|\begin{array}{c}\hat{\mathbf{i}}\hat{\mathbf{j}}\hat{\mathbf{k}}\\ a_1{a}_2{a}_3\\ b_1{b}_2{b}_3\end{array}\right|=⟨\hat{\mathbf{i}}\left|\begin{array}{c}{a}_2{a}_3\\ b_2{b}_3\end{array}\right|,-\hat{\mathbf{j}}\left|\begin{array}{c}{a}_1{a}_3\\ b_1{b}_3\end{array}\right|,\hat{\mathbf{k}}\left|\begin{array}{c}{a}_1{a}_2\\ b_1{b}_2\end{array}\right|⟩ \end{gathered}$$
因为叉积的结果是向量，所以$\mathbf{A}\times \mathbf{B}$和$\mathbf{B}\times \mathbf{A}$的结果并不相同，事实上：
$$\mathbf{A}\times \mathbf{B}=-(\mathbf{B}\times \mathbf{A})$$
NumPy 中可以通过cross函数来计算向量的叉积，代码如下所示。

print(np.cross(u, m1))  # [-14   7  21]
print(np.cross(m1, u))  # [ 14  -7 -21]

行列式

行列式（determinant）通常记作$det(\mathbf{A})$或$|\mathbf{A}|$，其中$\mathbf{A}$是一个$n$阶方阵。行列式可以看做是有向面积或体积的概念在一般欧几里得空间的推广，或者说行列式描述的是一个线性变换对“体积”所造成的影响。行列式的概念最早出现在解线性方程组的过程中，十七世纪晚期，关孝和（日本江户时代的数学家）与莱布尼茨的著作中已经使用行列式来确定线性方程组解的个数以及形式；十八世纪开始，行列式开始作为独立的数学概念被研究；十九世纪以后，行列式理论进一步得到发展和完善。

行列式的性质

行列式是由向量引出的，所以行列式解释的其实是向量的性质。

性质1：如果$det(\mathbf{A})$中某行（或某列）的元素全部为0，那么$det(\mathbf{A})=0$。

性质2：如果$det(\mathbf{A})$中某行（或某列）有公共因子$k$，则可以提出$k$，得到行列式$det({\mathbf{A}}^{{}^{\prime }})$，且$det(\mathbf{A})=k\cdot det({\mathbf{A}}^{{}^{\prime }})$。
$$det(\mathbf{A})=\left|\begin{array}{cccc}{a}_{11}& a_{12}& \dots & a_{1n}\\ ⋮& ⋮& \ddots & ⋮\\ k{a}_{i1}& k{a}_{i2}& \dots & k{a}_{in}\\ ⋮& ⋮& \ddots & ⋮\\ a_{n1}& a_{n2}& \dots & a_{nn}\end{array}\right|=k\left|\begin{array}{cccc}{a}_{11}& a_{12}& \dots & a_{1n}\\ ⋮& ⋮& \ddots & ⋮\\ a_{i1}& a_{i2}& \dots & a_{in}\\ ⋮& ⋮& \ddots & ⋮\\ a_{n1}& a_{n2}& \dots & a_{nn}\end{array}\right|=k\cdot det({\mathbf{A}}^{{}^{\prime }})$$

性质3：如果$det(\mathbf{A})$中某行（或某列）的每个元素是两数之和，则此行列式可拆分为两个行列式相加，如下所示。
$$\left|\begin{array}{cccc}{a}_{11}& a_{12}& \dots & a_{1n}\\ ⋮& ⋮& \ddots & ⋮\\ a_{i1}+b_{i1}& a_{i2}+b_{i2}& \dots & a_{in}+b_{in}\\ ⋮& ⋮& \ddots & ⋮\\ a_{n1}& a_{n2}& \dots & a_{nn}\end{array}\right|=\left|\begin{array}{cccc}{a}_{11}& a_{12}& \dots & a_{1n}\\ ⋮& ⋮& \ddots & ⋮\\ a_{i1}& a_{i2}& \dots & a_{in}\\ ⋮& ⋮& \ddots & ⋮\\ a_{n1}& a_{n2}& \dots & a_{nn}\end{array}\right|+\left|\begin{array}{cccc}{a}_{11}& a_{12}& \dots & a_{1n}\\ ⋮& ⋮& \ddots & ⋮\\ b_{i1}& b_{i2}& \dots & b_{in}\\ ⋮& ⋮& \ddots & ⋮\\ a_{n1}& a_{n2}& \dots & a_{nn}\end{array}\right|$$
性质4：如果$det(\mathbf{A})$中两行（或两列）元素对应成比例，那么$det(\mathbf{A})=0$。

性质5：如果$det(\mathbf{A})$中两行（或两列）互换得到$det({\mathbf{A}}^{{}^{\prime }})$，那么$det(\mathbf{A})=-det({\mathbf{A}}^{{}^{\prime }})$。

性质6：将$det(\mathbf{A})$中某行（或某列）的$k$倍加进另一行（或另一列）里，行列式的值不变，如下所示。
$$\left|\begin{array}{cccc}⋮& ⋮& ⋮& ⋮\\ a_{i1}& a_{i2}& \dots & a_{in}\\ a_{j1}& a_{j2}& \dots & a_{jn}\\ ⋮& ⋮& ⋮& ⋮\end{array}\right|=\left|\begin{array}{cccc}⋮& ⋮& ⋮& ⋮\\ a_{i1}& a_{i2}& \dots & a_{in}\\ a_{j1}+k{a}_{i1}& a_{j2}+k{a}_{i2}& \dots & a_{jn}+k{a}_{in}\\ ⋮& ⋮& ⋮& ⋮\end{array}\right|$$
性质7：将行列式的行列互换，行列式的值不变，如下所示。
$$\left|\begin{array}{cccc}{a}_{11}& a_{12}& \dots & a_{1n}\\ a_{21}& a_{22}& \dots & a_{2n}\\ ⋮& ⋮& \ddots & ⋮\\ a_{n1}& a_{n2}& \dots & a_{nn}\end{array}\right|=\left|\begin{array}{cccc}{a}_{11}& a_{21}& \dots & a_{n1}\\ a_{12}& a_{22}& \dots & a_{n2}\\ ⋮& ⋮& \ddots & ⋮\\ a_{1n}& a_{2n}& \dots & a_{nn}\end{array}\right|$$
性质8：方块矩阵$\mathbf{A}$和$\mathbf{B}$的乘积的行列式等于其行列式的乘积，即$det(\mathbf{A}\mathbf{B})=det(\mathbf{A})det(\mathbf{B})$。特别的，若将矩阵中的每一行都乘以常数$r$，那么行列式的值将是原来的$r^n$倍，即$det(r\mathbf{A})=det(r{\mathbf{I}}_{\mathbf{n}}\cdot \mathbf{A})=r^{n}det(\mathbf{A})$，其中${\mathbf{I}}_{\mathbf{n}}$是$n$阶单位矩阵。

性质9：若$\mathbf{A}$是可逆矩阵，那么$det({\mathbf{A}}^{-1})=(det(\mathbf{A}){)}^{-1}$。

行列式的计算

$n$阶行列式的计算公式如下所示：
$$det(\mathbf{A})=\sum _{n!}\pm a_{1\alpha }{a}_{2\beta }\cdots a_{n\omega }$$

对于二阶行列式，上面的公式相当于：
$$\left|\begin{array}{c}{a}_{11}{a}_{12}\\ a_{21}{a}_{22}\end{array}\right|=a_{11}{a}_{22}-a_{12}{a}_{21}$$
对于三阶行列式，上面的计算公式相当于：
$$\left|\begin{array}{c}{a}_{11}{a}_{12}{a}_{13}\\ a_{21}{a}_{22}{a}_{23}\\ a_{31}{a}_{32}{a}_{33}\end{array}\right|=a_{11}{a}_{22}{a}_{33}+a_{12}{a}_{23}{a}_{31}+a_{13}{a}_{21}{a}_{32}-a_{11}{a}_{23}{a}_{32}-a_{12}{a}_{21}{a}_{33}-a_{13}{a}_{22}{a}_{31}$$
高阶行列式可以用代数余子式（cofactor）展开成多个低阶行列式，如下所示：
$$det(\mathbf{A})=a_{11}{C}_{11}+a_{12}{C}_{12}+\cdots +a_{1n}{C}_{1n}=\sum _{i=1}^n{a}_{1i}{C}_{1i}$$
其中，$C_{11}$是原行列式去掉$a_{11}$所在行和列之后剩余的部分构成的行列式，以此类推。

矩阵

矩阵（matrix）是由一系列元素排成的矩形阵列，矩阵里的元素可以是数字、符号或数学公式。矩阵可以进行加法、减法、数乘、转置、矩阵乘法等运算，如下图所示。

值得一提的是矩阵乘法运算，该运算仅当第一个矩阵$\mathbf{A}$的列数和另一个矩阵$\mathbf{B}$的行数相等时才能定义。如果$\mathbf{A}$是一个$m\times n$的矩阵，$\mathbf{B}$是一个$n\times k$矩阵，它们的乘积是一个$m\times k$的矩阵，其中元素的计算公式如下所示：
$$[\mathbf{AB}{]}_{i,j}=A_{i,1}{B}_{1,j}+A_{i,2}{B}_{2,j}+\cdots +A_{i,n}{B}_{n,j}=\sum _{r=1}^n{A}_{i,r}{B}_{r,j}$$

矩阵的乘法满足结合律和对矩阵加法的分配律：

结合律： $(\mathbf{AB})\mathbf{C}=\mathbf{A}(\mathbf{BC})$。

左分配律：$(\mathbf{A}+\mathbf{B})\mathbf{C}=\mathbf{AC}+\mathbf{BC}$。

右分配律：$\mathbf{C}(\mathbf{A}+\mathbf{B})=\mathbf{CA}+\mathbf{CB}$。

矩阵乘法不满足交换律。一般情况下，矩阵$\mathbf{A}$和$\mathbf{B}$的乘积$\mathbf{AB}$存在，但$\mathbf{BA}$不一定存在，即便$\mathbf{BA}$存在，大多数时候$\mathbf{AB}\ne \mathbf{BA}$。

矩阵乘法的一个基本应用是在线性方程组上。线性方程组是方程组的一种，它符合以下的形式：
$$\{\begin{array}{l}{a}_{1,1}{x}_1+a_{1,2}{x}_2+\cdots +a_{1,n}{x}_n=b_1\\ a_{2,1}{x}_1+a_{2,2}{x}_2+\cdots +a_{2,n}{x}_n=b_2\\ ⋮⋮\\ a_{m,1}{x}_1+a_{m,2}{x}_2+\cdots +a_{m,n}{x}_n=b_m\end{array}$$
运用矩阵的方式，可以将线性方程组写成一个向量方程：
$$\mathbf{Ax}=\mathbf{b}$$
其中，$\mathbf{A}$是由方程组里未知数的系数排成的$m\times n$矩阵，$\mathbf{x}$是含有$n$个元素的行向量，$\mathbf{b}$是含有$m$个元素的行向量。

矩阵是线性变换（保持向量加法和标量乘法的函数）的便利表达法。矩阵乘法的本质在联系到线性变换的时候最能体现，因为矩阵乘法和线性变换的合成有以下的联系，即每个$m\times n$的矩阵$\mathbf{A}$都代表了一个从${\mathbf{R}}^n$射到${\mathbf{R}}^m$的线性变换。如果无法理解上面这些内容，推荐大家看看B站上名为《线性代数的本质》的视频，相信这套视频会让你对线性代数有一个更好的认知。

下图是一个来自于维基百科的例子，图中展示了一些典型的二维实平面上的线性变换对平面向量（图形）造成的效果以及它们对应的二维矩阵，其中每个线性变换将蓝色图形映射成绿色图形；平面的原点$(0,0)$用黑点表示。

矩阵对象

NumPy 中提供了专门用于线性代数（linear algebra）的模块和表示矩阵的类型matrix，当然我们通过二维数组也可以表示一个矩阵，官方并不推荐使用matrix类而是建议使用二维数组，而且有可能在将来的版本中会移除matrix类。无论如何，利用这些已经封装好的类和函数，我们可以轻松愉快的实现很多对矩阵的操作。

我们可以通过下面的代码来创建矩阵（matrix）对象。

代码：

m1 = np.matrix('1 2 3; 4 5 6')
m1

说明：matrix构造器可以传入类数组对象也可以传入字符串来构造矩阵对象。

输出：

matrix([[1, 2, 3],
        [4, 5, 6]])

代码：

m2 = np.asmatrix(np.array([[1, 1], [2, 2], [3, 3]]))
m2

说明：asmatrix函数也可以用mat函数代替，这两个函数其实是同一个函数。

输出：

matrix([[1, 1],
        [2, 2],
        [3, 3]])

代码：

m1 * m2

输出：

matrix([[14, 14],
        [32, 32]])

说明：注意matrix对象和ndarray对象乘法运算的差别，matrix对象的*运算是矩阵乘法运算。如果两个二维数组要做矩阵乘法运算，应该使用@运算符或matmul函数，而不是*运算符。

矩阵对象的属性如下表所示。

属性	说明
A	获取矩阵对象对应的ndarray对象
A1	获取矩阵对象对应的扁平化后的ndarray对象
I	可逆矩阵的逆矩阵
T	矩阵的转置
H	矩阵的共轭转置
shape	矩阵的形状
size	矩阵元素的个数

矩阵对象的方法跟之前讲过的ndarray数组对象的方法基本差不多，此处不再进行赘述。

线性代数模块

NumPy 的linalg模块中有一组标准的矩阵分解运算以及诸如求逆和行列式之类的函数，它们跟 MATLAB 和 R 等语言所使用的是相同的行业标准线性代数库，下面的表格列出了numpy以及linalg模块中一些常用的线性代数相关函数。

函数	说明
diag	以一维数组的形式返回方阵的对角线元素或将一维数组转换为方阵（非对角元素元素为0）
matmul	矩阵乘法运算
trace	计算对角线元素的和
norm	求矩阵或向量的范数
det	计算行列式的值
matrix_rank	计算矩阵的秩
eig	计算矩阵的特征值（eigenvalue）和特征向量（eigenvector）
inv	计算非奇异矩阵（$n$阶方阵）的逆矩阵
pinv	计算矩阵的摩尔-彭若斯（Moore-Penrose）广义逆
qr	QR分解（把矩阵分解成一个正交矩阵与一个上三角矩阵的积）
svd	计算奇异值分解（singular value decomposition）
solve	解线性方程组$\mathbf{Ax}=\mathbf{b}$，其中$\mathbf{A}$是一个方阵
lstsq	计算$\mathbf{Ax}=\mathbf{b}$的最小二乘解

下面我们简单尝试一下上面的函数，先试一试求逆矩阵。

代码：

m3 = np.array([[1., 2.], [3., 4.]])
m4 = np.linalg.inv(m3)
m4

输出：

array([[-2. ,  1. ],
       [ 1.5, -0.5]])

代码：

np.around(m3 @ m4)

说明：around函数对数组元素进行四舍五入操作，默认小数点后面的位数为0。

输出：

array([[1., 0.],
       [0., 1.]])

说明：矩阵和它的逆矩阵做矩阵乘法会得到单位矩阵。

计算行列式的值。

代码：

m5 = np.array([[1, 3, 5], [2, 4, 6], [4, 7, 9]])
np.linalg.det(m5)

输出：

2

计算矩阵的秩。

代码：

np.linalg.matrix_rank(m5)

输出：

3

求解线性方程组。
$$\{\begin{array}{l}{x}_1+2x_2+x_3=8\\ 3x_1+7x_2+2x_3=23\\ 2x_1+2x_2+x_3=9\end{array}$$

对于上面的线性方程组，我们可以用矩阵的形式来表示它，如下所示。
$$\mathbf{A}=\left[\begin{array}{ccc}1& 2& 1\\ 3& 7& 2\\ 2& 2& 1\end{array}\right],\mathbf{x}=\left[\begin{array}{c}{x}_1\\ x_2\\ x_3\end{array}\right],\mathbf{b}=\left[\begin{array}{c}8\\ 23\\ 9\end{array}\right]$$

$$\mathbf{Ax}=\mathbf{b}$$

线性方程组有唯一解的条件：系数矩阵$\mathbf{A}$的秩等于增广矩阵$\mathbf{Ab}$的秩，而且跟未知数的个数相同。

代码：

A = np.array([[1, 2, 1], [3, 7, 2], [2, 2, 1]])
b = np.array([8, 23, 9]).reshape(-1, 1)
print(np.linalg.matrix_rank(A))
print(np.linalg.matrix_rank(np.hstack((A, b))))

说明：使用数组对象的reshape方法调形时，如果其中一个参数为-1，那么该维度有多少个元素是通过数组元素个数（size属性）和其他维度的元素个数自动计算出来的。

输出：

3
3

代码：

np.linalg.solve(A, b)

输出：

array([[1.],
       [2.],
       [3.]])

说明：上面的结果表示，线性方程组的解为：$x_1=1,x_2=2,x_3=3$。

下面是另一种求解线性方程组的方法，大家可以停下来思考下为什么。
$$\mathbf{x}={\mathbf{A}}^{-1}\cdot \mathbf{b}$$
代码：

np.linalg.inv(A) @ b

输出：

array([[1.],
       [2.],
       [3.]])

多项式

除了数组，NumPy 中还封装了用于多项式（polynomial）运算的数据类型。多项式是变量的整数次幂与系数的乘积之和，形如：
$$f(x)=a_n{x}^n+a_{n-1}{x}^{n-1}+\cdots +a_1{x}^1+a_0{x}^0$$
在 NumPy 1.4版本之前，我们可以用poly1d类型来表示多项式，目前它仍然可用，但是官方提供了新的模块numpy.polynomial，它除了支持基本的幂级数多项式外，还可以支持切比雪夫多项式、拉盖尔多项式等。

创建多项式对象

创建poly1d对象，例如：$f(x)=3x^2+2x+1$。

代码：

p1 = np.poly1d([3, 2, 1])
p2 = np.poly1d([1, 2, 3])
print(p1)
print(p2)

输出：

   2
3 x + 2 x + 1
   2
1 x + 2 x + 3

多项式的操作

获取多项式的系数

代码：

print(p1.coefficients)
print(p2.coeffs)

输出：

[3 2 1]
[1 2 3]

两个多项式的四则运算

代码：

print(p1 + p2)
print(p1 * p2)

输出：

   2
4 x + 4 x + 4
   4     3      2
3 x + 8 x + 14 x + 8 x + 3

带入$x$求多项式的值

代码：

print(p1(3))
print(p2(3))

输出：

34
18

多项式求导和不定积分

代码：

print(p1.deriv())
print(p1.integ())

输出：

6 x + 2
   3     2
1 x + 1 x + 1 x

求多项式的根

例如有多项式$f(x)=x^2+3x+2$，多项式的根即一元二次方程$x^2+3x+2=0$的解。

代码：

p3 = np.poly1d([1, 3, 2])
print(p3.roots)

输出：

[-2. -1.]

如果使用numpy.polynomial模块的Polynomial类来表示多项式对象，那么对应的操作如下所示。

代码：

from numpy.polynomial import Polynomial

p3 = Polynomial((2, 3, 1))
print(p3)           # 输出多项式
print(p3(3))        # 令x=3，计算多项式的值
print(p3.roots())   # 计算多项式的根
print(p3.degree())  # 获得多项式的次数
print(p3.deriv())   # 求导
print(p3.integ())   # 求不定积分

输出：

2.0 + 3.0·x + 1.0·x²
20.0
[-2. -1.]
2
3.0 + 2.0·x
0.0 + 2.0·x + 1.5·x² + 0.33333333·x³

最小二乘解

Polynomial类还有一个名为fit的类方法，它可以给多项式求最小二乘解。所谓最小二乘解（least-squares solution），是用最小二乘法通过最小化误差的平方和来寻找数据的最佳匹配函数的系数。假设多项式为$f(x)=ax+b$，最小二乘解就是让下面的残差平方和$RSS$达到最小的$a$和$b$。
$$RSS=\sum _{i=0}^k(f(x_i)-y_i{)}^2$$
例如，我们想利用收集到的月收入和网购支出的历史数据来建立一个预测模型，以达到通过某人的月收入预测他网购支出金额的目标，下面是我们收集到的收入和网购支出的数据，保存在两个数组中。

x = np.array([
    25000, 15850, 15500, 20500, 22000, 20010, 26050, 12500, 18500, 27300,
    15000,  8300, 23320,  5250,  5800,  9100,  4800, 16000, 28500, 32000,
    31300, 10800,  6750,  6020, 13300, 30020,  3200, 17300,  8835,  3500
])
y = np.array([
    2599, 1400, 1120, 2560, 1900, 1200, 2320,  800, 1650, 2200,
     980,  580, 1885,  600,  400,  800,  420, 1380, 1980, 3999,
    3800,  725,  520,  420, 1200, 4020,  350, 1500,  560,  500
])

我们可以先绘制散点图来了解两组数据是否具有正相关或负相关关系。正相关意味着数组x中较大的值对应到数组y中也是较大的值，而负相关则意味着数组x中较大的值对应到数组y中较小的值。

import matplotlib.pyplot as plt

plt.figure(dpi=120)
plt.scatter(x, y, color='blue')
plt.show()

输出：

如果需要定量的研究两组数据的相关性，我们可以计算协方差或相关系数，对应的 NumPy 函数分别是cov和corrcoef。

代码：

np.corrcoef(x, y)

输出：

array([[1.        , 0.92275889],
       [0.92275889, 1.        ]])

说明：相关系数是一个-1到1之间的值，越靠近1 说明正相关性越强，越靠近-1说明负相关性越强，靠近0则说明两组数据没有明显的相关性。上面月收入和网购支出之间的相关系数是0.92275889，说明二者是强正相关关系。

通过上面的操作，我们确定了收入和网购支出之前存在强正相关关系，于是我们用这些数据来创建一个回归模型，找出一条能够很好的拟合这些数据点的直线。这里，我们就可以用到上面提到的fit方法，具体的代码如下所示。

代码：

from numpy.polynomial import Polynomial

Polynomial.fit(x, y, deg=1).convert().coef

说明：deg=1说明回归模型最高次项就是1次项，回归模型形如$y=ax+b$；如果要生一个类似于$y=a{x}^2+bx+c$的模型，就需要设置deg=2，以此类推。

输出：

array([-2.94883437e+02,  1.10333716e-01])

根据上面输出的结果，我们的回归方程应该是$y=0.110333716x-294.883437$。我们将这个回归方程绘制到刚才的散点图上，红色的点是我们的预测值，蓝色的点是历史数据，也就是真实值。

代码：

import matplotlib.pyplot as plt

plt.scatter(x, y, color='blue')
plt.scatter(x, 0.110333716 * x - 294.883437, color='red')
plt.plot(x, 0.110333716 * x - 294.883437, color='darkcyan')
plt.show()

输出：

如果不使用Polynomial类型的fit方法，我们也可以通过 NumPy 提供的polyfit函数来完成同样的操作，有兴趣的读者可以自行研究。