线性代数
线性代数(如矩阵乘法、矩阵分解、行列式以及其他方阵数学等)是任何数组库的重要组成部分。不像某些语言(如MATLAB),通过*对两个二维数组相乘得到的是一个元素级的积,而不是一个矩阵点积。因此,NumPy提供了一个用于矩阵乘法的dot函数(既是一个数组方法也是numpy命名空间中的一个函数):
- In [194]: x = np.array([[1., 2., 3.], [4., 5., 6.]])
- In [195]: y = np.array([[6., 23.], [-1, 7], [8, 9]])
- In [196]: x In [197]: y
- Out[196]: Out[197]:
- array([[ 1., 2., 3.], array([[ 6., 23.],
- [ 4., 5., 6.]]) [ -1., 7.],
- [ 8., 9.]])
- In [198]: x.dot(y) # 相当于np.dot(x, y)
- Out[198]:
- array([[ 28., 64.],
- [ 67., 181.]])
一个二维数组跟一个大小合适的一维数组的矩阵点积运算之后将会得到一个一维数组:
- In [199]: np.dot(x, np.ones(3))
- Out[199]: array([ 6., 15.])
numpy.linalg中有一组标准的矩阵分解运算以及诸如求逆和行列式之类的东西。它们跟MATLAB和R等语言所使用的是相同的行业标准级Fortran库,如BLAS、LAPACK、Intel MKL(可能有,取决于你的NumPy版本)等:
- In [201]: from numpy.linalg import inv, qr
- In [202]: X = randn(5, 5)
- In [203]: mat = X.T.dot(X)
- In [204]: inv(mat)
- Out[204]:
- array([[ 3.0361, -0.1808, -0.6878, -2.8285, -1.1911],
- [-0.1808, 0.5035, 0.1215, 0.6702, 0.0956],
- [-0.6878, 0.1215, 0.2904, 0.8081, 0.3049],
- [-2.8285, 0.6702, 0.8081, 3.4152, 1.1557],
- [-1.1911, 0.0956, 0.3049, 1.1557, 0.6051]])
- In [205]: mat.dot(inv(mat))
- Out[205]:
- array([[ 1., 0., 0., 0., -0.],
- [ 0., 1., -0., 0., 0.],
- [ 0., -0., 1., 0., 0.],
- [ 0., -0., -0., 1., -0.],
- [ 0., 0., 0., 0., 1.]])
- In [206]: q, r = qr(mat)
- In [207]: r
- Out[207]:
- array([[ -6.9271, 7.389 , 6.1227, -7.1163, -4.9215],
- [ 0. , -3.9735, -0.8671, 2.9747, -5.7402],
- [ 0. , 0. , -10.2681, 1.8909, 1.6079],
- [ 0. , 0. , 0. , -1.2996, 3.3577],
- [ 0. , 0. , 0. , 0. , 0.5571]])
表4-7中列出了一些最常用的线性代数函数。
注意: Python科学计算社区盼望着有朝一日能实现矩阵乘法的中缀运算符,以便能用一种更漂亮的语法代替np.dot。不过目前就只能先这样了。