Python中的Numpy包提供了强大的矩阵运算能力,下面我们简单的介绍一下这些运算的代码,让大家能够顺利地使用numpy去实现这些基本运算。除此之外我还将介绍一些矩阵基本概念的通俗理解,持续更新中……
首先我们要导入numpy这个包,两种导入方式:
In [1]: from numpy import *
In [2]: import numpy as py
个人建议使用第二种方式,这里由于我们要演示绝大部分的操作,所以就不一一使用np.去调用那些方法了,而采用导入全部库函数的方式。
首先矩阵是什么,我们就不做过多的介绍了,让我们看看numpy是怎么创建矩阵的:
两种创建方式,直接创建,或者是先创建数组,再转换成矩阵,它们之间是等价的
In [3]: a1 = array([1,2,3])
In [5]: a1
Out[5]: array([1, 2, 3])
In [6]: a1 = mat(a1)
In [7]: a1
Out[7]: matrix([[1, 2, 3]])
In [8]: shape(a1)
Out[8]: (1, 3)
In [9]: b = matrix([1,2,3])
In [10]: b
Out[10]: matrix([[1, 2, 3]])
In [11]: shape(b)
Out[11]: (1, 3)
有很多常见的矩阵shape很大,我们不可能一个一个去输入吧,比如一个10*10的零矩阵,如果我们一个一个去输入那肯定是不靠谱的,不用担心,numpy提供了创建这些矩阵的函数,我们所要做的就只是输入我们的shape就可以啦。
In [17]: data1 = zeros(shape=(3,3))
In [18]: data1
Out[18]:
array([[ 0., 0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]])
In [19]: data1 = mat(data1)
In [20]: data1
Out[20]:
matrix([[ 0., 0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]])
In [21]: data1=mat(zeros((3,3)))
In [22]: data1
Out[22]:
matrix([[ 0., 0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]])
In [28]: data2=mat(ones(shape=(2,4),dtype=int)) #可以选择你想要的任意数据类型
In [29]: data2
Out[29]:
matrix([[1, 1, 1, 1],
[1, 1, 1, 1]])
In [31]: data3 = mat(random.rand(2,2))
In [32]: data3
Out[32]:
matrix([[ 0.45544031, 0.02462429],
[ 0.8394628 , 0.38270193]])
In [33]: data4 = mat(random.randint(10,size=(3,3))) #生成0-10之间3*3的随机矩阵
In [34]: data4
Out[34]:
matrix([[8, 3, 7],
[1, 0, 6],
[7, 5, 0]])
In [35]: data5 = eye(2,2,dtype=float) #生成相应的对角矩阵
In [36]: data5
Out[36]:
array([[ 1., 0.],
[ 0., 1.]])
In [37]: data5 = mat(data5)
In [38]: data5
Out[38]:
matrix([[ 1., 0.],
[ 0., 1.]])
In [39]: a1 = [1,2,3]
In [40]: a2 = mat((diag(a1))) #生成【1,2,3】的对角矩阵
In [41]: a2
Out[41]:
matrix([[1, 0, 0],
[0, 2, 0],
[0, 0, 3]])
下面介绍常见的矩阵运算:
1、矩阵相乘
In [42]: a1 = mat([1,2])
In [43]: a2 = mat([[1],[2]])
In [44]: a3 = a1*a2
In [45]: a3
Out[45]: matrix([[5]])
2、矩阵点乘
#元素点乘
In [46]: a1 = mat([1,1])
In [47]: a2 = mat([2,2])
In [48]: a3 = multiply(a1,a2)
In [49]: a3
Out[49]: matrix([[2, 2]])
In [50]: a1 = mat([2,2])
In [51]: a2 = a1*2
In [52]: a2
Out[52]: matrix([[4, 4]])
3、矩阵求逆和转置
求逆
In [56]: a1 = mat(eye(2,2)*0.5)
In [57]: a1
Out[57]:
matrix([[ 0.5, 0. ],
[ 0. , 0.5]])
In [58]: a2 = a1.I
In [59]: a2
Out[59]:
matrix([[ 2., 0.],
[ 0., 2.]])
转置
In [60]: a1 = mat([[1,1],[0,0]])
In [61]: a2 = a1.T
In [62]: a2
Out[62]:
matrix([[1, 0],
[1, 0]])
4、计算对应行列的最大值最小值,和
计算对应行列的和
In [63]: a1 = mat([[1,1],[2,4],[3,5]])
In [64]: a2 = a1.sum(axis=0)
In [65]: a2
Out[65]: matrix([[ 6, 10]])
In [66]: a3 = a1.sum(axis=1)
In [67]: a3
Out[67]:
matrix([[2],
[6],
[8]])
In [68]: a1.max() #计算全部元素的最大值
Out[68]: 5
In [69]: a2=max(a1[:,1]) #计算第二列的最大值,这里得到的是一个1*1的矩阵
In [70]: a2
Out[70]: matrix([[5]])
In [72]: a1[1,:].max() #计算第二行的最大值,这里得到的是一个一个数值
Out[72]: 4
In [75]: np.max(a1,0) #计算所有列的最大值,这里使用的是numpy中的max函数,注意需要导入 import numpy as np,主要是为了区别一般的max函数
Out[75]: matrix([[3, 5]])
In [76]: np.max(a1,1)#计算所有行的最大值,这里得到是一个矩阵
Out[76]:
matrix([[1],
[4],
[5]])
In [77]: np.argmax(a1,0)#计算所有列的最大值对应在该列中的索引
Out[77]: matrix([[2, 2]])
In [78]: np.argmax(a1[1,:])#计算第二行中最大值对应在该行的索引
Out[78]: 1
5、矩阵的分割和合并
In [79]: a = mat(ones((3,3)))
In [80]: a
Out[80]:
matrix([[ 1., 1., 1.],
[ 1., 1., 1.],
[ 1., 1., 1.]])
In [81]: b = a[1:,1:] #分割出第二行以后的行和第二列以后的列的所有元素
In [82]: b
Out[82]:
matrix([[ 1., 1.],
[ 1., 1.]])
In [83]: a = mat(ones((2,2)))
In [84]: b = mat(eye(2))
In [85]: c = vstack((a,b))#按列合并,即增加行数
In [86]: c
Out[86]:
matrix([[ 1., 1.],
[ 1., 1.],
[ 1., 0.],
[ 0., 1.]])
In [87]: d = hstack((a,b))#按行合并,即行数不变,扩展列数
In [88]: d
Out[88]:
matrix([[ 1., 1., 1., 0.],
[ 1., 1., 0., 1.]])
6、求解矩阵的迹
In [89]: E=np.array([[1,2,3],[4,5,6],[7,8,9]])
In [90]: E
Out[90]:
array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
In [91]: np.trace(E)
Out[91]: 15