NumPy(Numerical Python)是一个用于科学计算第三方的Python包。

 NumPy提供了许多高级的数值编程工具,如:矩阵数据类型、矢量处理,以及精密的运算库。专为进行严格的数字处理而产生。


一、数组简介


Numpy中,最重要的数据结构是:多维数组类型(numpy.ndarray)

ndarray由两部分组成:

实际所持有的数据;

描述这些数据的元数据(metadata)

数组(即矩阵)的维度被称为axes,维数称为rank

ndarray 的重要属性包括:


        ndarray.ndim:数组的维数,也称为rank


        ndarray.shape:数组各维的大小,对一个n行m列的矩阵来说, shape 为 (n,m)


        ndarray.size:元素的总数。


        ndarray.dtype:每个元素的类型,可以是numpy.int32, numpy.int16, and numpy.float64等


        ndarray.itemsize:每个元素占用的字节数。


        ndarray.data:指向数据内存。


二、数组的使用


使用numpy前要先导入模块,使用下面的语句导入模块:

improt numpy as np #其中np为numpy的别名,是一种习惯用法

1.使用array方法生成数组

       array,也就是数组,是numpy中最基础的数据结构,最关键的属性是维度和元素类型,在numpy中,可以非常方便地创建各种不同类型的多维数组,并且执行一些基本基本操作,生成数组的方法有一下几种:

以list或tuple变量产生以为数组:

>>> print np.array([1,2,3,4])
[1 2 3 4]
>>> print np.array((1.2,2,3,4))
[ 1.2  2.   3.   4. ]

以list或tuple变量为元素产生二维数组或者多维数组:

>>> x = np.array(((1,2,3),(4,5,6)))
>>> x
array([[1, 2, 3],
       [4, 5, 6]])
>>> y = np.array([[1,2,3],[4,5,6]])
>>> y
array([[1, 2, 3],
       [4, 5, 6]])

2.使用numpy.arange方法生成数组

>>> print np.arange(15)
[ 0  1  2  3  4  5  6  7  8  9  10  11  12  13  14]
>>> print type(np.arange(15))
<type 'numpy.ndarray'>

3.使用内置函数生成特殊矩阵(数组)

零矩阵

>>> print np.zeros((3,4))
[[ 0.  0.  0.  0.]
 [ 0.  0.  0.  0.]
 [ 0.  0.  0.  0.]]

一矩阵


>>> print np.ones((3,4))
[[ 1. 1.  1.  1.]
 [1.  1. 1.  1.]
 [ 1.  1. 1.  1.]]

单位矩阵

>>> print np.eye(3)
[[ 1. 0.  0.]
 [0.  1. 0.]
 [ 0.  0. 1.]]

4.索引与切片

>>> x = np.array(((1,2,3),(4,5,6)))
>>> x[1,2] #获取第二行第三列的数
6
>>> y=x[:,1] #获取第二列
>>> y
array([2, 5])

与python语法一致,不再举例。

5.获取数组属性

>>> a = np.zeros((2,2,2))
>>> print a.ndim   #数组的维数
3
>>> print a.shape  #数组每一维的大小
(2, 2, 2)
>>> print a.size   #数组的元素数
8
>>> print a.dtype  #元素类型
float64
>>> print a.itemsize  #每个元素所占的字节数
8

6.数组变换

多维转换为一维:

>>> x
array([[1, 2, 3],
       [4, 5, 6]])
>>> x.flatten()
array([1, 2, 3, 4, 5, 6])

一维转换为多维:

>>> print np.arange(15).reshape(3,5) #改变形状,将一维的改成三行五列
[[ 0  1  2  3  4]
 [ 5  6  7  8  9]
 [10 11 12 13 14]]

转置:

>>> x
array([[1, 2, 3],
       [4, 5, 6]])
>>> x.transpose()
array([[1, 4],
       [2, 5],
       [3, 6]])

7.数组组合

水平组合:

>>> y=x
>>> numpy.hstack((x,y))
array([[1, 2, 3, 1, 2, 3],
       [4, 5, 6, 4, 5, 6]]

垂直组合

>>> numpy.vstack((x,y))
array([[1, 2, 3],
       [4, 5, 6],
       [1, 2, 3],
       [4, 5, 6]])

用concatenate函数可以同时实现这两种方式,通过指定axis参数,默认为0,垂直组合。


>>> numpy.concatenate((x,y))
array([[1, 2, 3],
       [4, 5, 6],
       [1, 2, 3],
       [4, 5, 6]])
>>> numpy.concatenate((x,y),axis=1)
array([[1, 2, 3, 1, 2, 3],
       [4, 5, 6, 4, 5, 6]])

8.数组分割

垂直分割

>>> z
array([[1, 2, 3],
       [4, 5, 6],
       [1, 2, 3],
       [4, 5, 6]])
>>> numpy.vsplit(z,2) #注意这里设置的分割数目必须可以被行数整除
[array([[1, 2, 3],
       [4, 5, 6]]), array([[1, 2, 3],
       [4, 5, 6]])]

水平分割

>>> numpy.hsplit(z,3)
[array([[1],
       [4],
       [1],
       [4]]), array([[2],
       [5],
       [2],
       [5]]), array([[3],
       [6],
       [3],
       [6]])]

用split函数可以同时实现这两个效果,通过设置其axis参数区别,与组合类似,这里不在演示。


三、矩阵


通过上面对数组的操作可以知道,numpy中可以通过数组模拟矩阵,但是numpy也有专门处理矩阵的数据结构——matrix。

1.生成矩阵

>>> numpy.mat('1 2 3;4 5 6;7 8 9')
matrix([[1, 2, 3],
        [4, 5, 6],
        [7, 8, 9]])

2.数组矩阵转化

矩阵转数组

>>> m=numpy.mat('1 2 3;4 5 6;7 8 9')
>>> numpy.array(m)
array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])

数组转矩阵

>>> n=numpy.array(m)
>>> numpy.mat(n)
matrix([[1, 2, 3],
        [4, 5, 6],
        [7, 8, 9]])

3.矩阵方法

求逆:

>>> m.I
matrix([[ -4.50359963e+15,   9.00719925e+15,  -4.50359963e+15],
        [  9.00719925e+15,  -1.80143985e+16,   9.00719925e+15],
        [ -4.50359963e+15,   9.00719925e+15,  -4.50359963e+15]])