NumPY是Python的一个矩阵处理库,提供了大量点矩阵处理函数,其内部通过C语言实现,执行速度快。
1.一维NumPY数组
使用NumPY提供的arange(n)函数可以创建一个包含n个元素的NumPY数组,取值分别为0~n-1.
>>> from NumPY import *
>>> a=arange(4)
>>> a
array([0, 1, 2, 3])
>>> a.shape
(4,)
>>>
使用array()创建NumPY数组.
>>> mm=array((4,5,1))
>>> mm
array([4, 5, 1])
>>> pp=array((10,20,30))
>>> pp
array([10, 20, 30])
>>> mm+pp
array([14, 25, 31])
>>> pp*2
array([20, 40, 60])
使用标准的Python时,向量运算需要循环实现,使用NumPY可以直接运算. 对每个元素平方:
>>> mm**2
array([16, 25, 1])
访问元素:
>>> mm[1]
5
>>> pp[2]
30
>>> 2*mm[1]
10
>>>
2. 多维NumPY数组
>>> jj=array([[1,2,3],[4,5,6],[7,8,9]])
>>> jj
array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
数组访问:
>>> jj[1]
array([4, 5, 6])
>>> jj[1][1]
5
>>> jj[0,2]
3
数组相乘
>>> a1=array([1,2,3])
>>> b1=array([4,8,10])
>>> a1*b1
array([ 4, 16, 30])
>>> a2=array([[2,3,4],[5,6,7]])
>>> b2=array([[1,2,3],[4,5,6]])
>>> a2*b2
array([[ 2, 6, 12],
[20, 30, 42]])
>>>
可以看到NumPY数组相乘,两个数组行和列都是相同的,然后对应位置相乘。数组乘法和矩阵乘法还是有区别的,矩阵乘法中两个矩阵必须满足m*n和n*k的条件。
3.NumPY矩阵
创建NumPY矩阵需要导入matrix或mat模块
>>> from numpy import mat,matrix
>>> m1=mat([[1,2,3],[2,3,1]])
>>> m1
matrix([[1, 2, 3],
[2, 3, 1]])
>>> m2=mat([[1,1,2],[2,1,1]])
>>> m2
matrix([[1, 1, 2],
[2, 1, 1]])
>>> m1*m2
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/Library/Frameworks/Python.framework/Versions/3.5/lib/python3.5/site-packages/numpy/matrixlib/defmatrix.py", line 343, in __mul__
return N.dot(self, asmatrix(other))
ValueError: shapes (2,3) and (2,3) not aligned: 3 (dim 1) != 2 (dim 0)
这时候m1*m2会发生错误,m1,m2 都为2*3的矩阵,因此不能相乘。把m2转置为3*2的矩阵:
>>> m2.T
matrix([[1, 2],
[1, 1],
[2, 1]])
>>> m1*m2.T
matrix([[9, 7],
[7, 8]])
如果想把矩阵m1和m2中的元素对应相乘,可以使用multiply函数:
>>> from numpy import multiply
>>> m1
matrix([[1, 2, 3],
[2, 3, 1]])
>>> m2
matrix([[1, 1, 2],
[2, 1, 1]])
>>> multiply(m1,m2)
matrix([[1, 2, 6],
[4, 3, 1]])
使用shape函数查看矩阵的维数:
>>> m1.shape
(2, 3)
使用sort函数可以把矩阵中的每一行元素进行排序:
>>> m3=mat([[7,5,4],[9,8,7],[5,2,0]])
>>> m3
matrix([[7, 5, 4],
[9, 8, 7],
[5, 2, 0]])
>>> m3.sort()
>>> m3
matrix([[4, 5, 7],
[7, 8, 9],
[0, 2, 5]])
计算矩阵平均值:
>>> m3.mean()
5.2222222222222223
取出某一行元素:
>>> m3
matrix([[4, 5, 7],
[7, 8, 9],
[0, 2, 5]])
>>> m3[1,:]
matrix([[7, 8, 9]])
>>> m3[2,:]
matrix([[0, 2, 5]])
取指定范围元素, 如取出矩阵m3中第三行中的第二列和第三列,也就算2和5:
>>> m3[2,1:3]
matrix([[2, 5]])
官方文档:numpy-1.10.1