天天看点

python工具——NumPy

NumPy(Numerical Python)是Python中科学计算的基础包。

它是一个Python库,提供多维数组对象,各种派生对象(如掩码数组和矩阵),以及用于数组快速操作的各种API,有包括数学、逻辑、形状操作、排序、选择、输入输出、离散傅立叶变换、基本线性代数,基本统计运算和随机模拟等等。

NumPy的主要对象是同构多维数组。

它是一个元素表(通常是数字),所有类型都相同,由非负整数元组索引。

在NumPy维度中称为轴 

numpy提供了python对多维数组对象的支持:ndarray

ndarray对象的属性

ndarray.ndim - 数组的轴(维度)的个数
ndarray.shape - 数组的维度
ndarray.size - 数组元素的总数
ndarray.dtype - 一个描述数组中元素类型的对象
ndarray.itemsize - 数组中每个元素的字节大小      

eg:

>>> import numpy as np
>>> a=np.arange(12).reshape(3,4)
>>> a
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])
>>> a.shape
(3, 4)
>>> a.ndim
2
>>> a.dtype.name
'int32'
>>> a.itemsize
4
>>> a.size
12
>>> type(a)
<class 'numpy.ndarray'>      

 数组创建

>>> b = np.array([2,3,4])
>>> b
array([2, 3, 4])
>>> b.dtype
dtype('int32')
>>> c=np.array([1.2,2.3,3.4])
>>> c.dtype
dtype('float64')
>>> d=np.array([(1.2,2,3),(4,5,6)])
>>> d
array([[1.2, 2. , 3. ],
       [4. , 5. , 6. ]])
>>> e=np.array([[1,2],[4,5]],dtype=complex)
>>> e
array([[1.+0.j, 2.+0.j],
       [4.+0.j, 5.+0.j]])      

索引、切片

>>> a
array([  0,   1,   8,  27,  64, 125, 216, 343, 512, 729], dtype=int32)
>>> a[2]
8
>>> a[2:5]
array([ 8, 27, 64], dtype=int32)
>>> a[:6:2]=-1000
>>> a
array([-1000,     1, -1000,    27, -1000,   125,   216,   343,   512,
         729], dtype=int32)
>>> a[ : :-1]
array([  729,   512,   343,   216,   125, -1000,    27, -1000,     1,
       -1000], dtype=int32)      
python工具——NumPy

与matplotlib

构建直方图

import numpy as np
import matplotlib.pyplot as plt
mu, sigma = 2, 0.5
v = np.random.normal(mu,sigma,10000)
plt.hist(v, bins=50, density=1)
plt.show()      
python工具——NumPy

pylab.hist

自动绘制直方图

import numpy as np
import matplotlib.pyplot as plt
mu, sigma = 2, 0.5
v = np.random.normal(mu,sigma,10000)
(n, bins) = np.histogram(v, bins=50, density=True)
plt.plot(.5*(bins[1:]+bins[:-1]), n)
plt.show()