Python中的Numpy包提供了強大的矩陣運算能力桐猬,下面我們簡單的介紹一下這些運算的代碼刽肠,讓大家能夠順利地使用numpy去實現(xiàn)這些基本運算。除此之外我還將介紹一些矩陣基本概念的通俗理解音五,持續(xù)更新中……
首先我們要導入numpy這個包,兩種導入方式:
In [1]: from numpy import *
In [2]: import numpy as py
個人建議使用第二種方式放仗,這里由于我們要演示絕大部分的操作,所以就不一一使用np.去調用那些方法了诞挨,而采用導入全部庫函數(shù)的方式。
首先矩陣是什么惶傻,我們就不做過多的介紹了,讓我們看看numpy是怎么創(chuàng)建矩陣的:
兩種創(chuàng)建方式银室,直接創(chuàng)建,或者是先創(chuàng)建數(shù)組蜈敢,再轉換成矩陣,它們之間是等價的
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提供了創(chuàng)建這些矩陣的函數(shù),我們所要做的就只是輸入我們的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)) #可以選擇你想要的任意數(shù)據(jù)類型
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() #計算第二行的最大值缎罢,這里得到的是一個一個數(shù)值
Out[72]: 4
In [75]: np.max(a1,0) #計算所有列的最大值,這里使用的是numpy中的max函數(shù)策精,注意需要導入 import numpy as np,主要是為了區(qū)別一般的max函數(shù)
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))#按列合并谜嫉,即增加行數(shù)
In [86]: c
Out[86]:
matrix([[ 1., 1.],
[ 1., 1.],
[ 1., 0.],
[ 0., 1.]])
In [87]: d = hstack((a,b))#按行合并,即行數(shù)不變沐兰,擴展列數(shù)
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