numpy学习:arangy函数和meshgrid函数

arange函数用于创建等差数组。

arange([start,] stop[, step,], dtype=None)根据start与stop指定的范围以及step设定的步长以及返回类型,生成一个 ndarray。

    >>> np.arange(3)  
      array([0, 1, 2])  
      >>> np.arange(3.0)  
      array([ 0.,  1.,  2.])  
      >>> np.arange(3,7)  
      array([3, 4, 5, 6])  
      >>> np.arange(3,7,2)  
      array([3, 5]) 


>>> arange(0,1,0.1)
array([ 0. ,  0.1,  0.2,  0.3,  0.4,  0.5,  0.6,  0.7,  0.8,  0.9])

meshgrid函数将两个输入的数组x和y进行扩展,前一个的扩展与后一个有关,后一个的扩展与前一个有关,前一个是竖向扩展,后一个是横向扩展。因为,y的大小为2,所以x竖向扩展为原来的两倍,而x的大小为3,所以y横向扩展为原来的3倍。通过meshgrid函数之后,输入由原来的数组变成了一个矩阵。通过使用meshgrid函数,可以产生一个表格矩阵,下面用一个例子来展示产生一个2*2网格的坐标,每个网格的大小为1。

    nx,ny = (3,2)  
    #从0开始到1结束,返回一个numpy数组,nx代表数组中元素的个数  
    x = np.linspace(0,1,nx)  
    #[ 0.   0.5  1. ]  
    y = np.linspace(0,1,ny)  
    # [0.  1.]  
    xv,yv = np.meshgrid(x,y)  
    ''''' 
    xv 
    [[ 0.   0.5  1. ] 
     [ 0.   0.5  1. ]] 
     yv 
     [[ 0.  0.  0.] 
      [ 1.  1.  1.]] 
    '''  

猜你喜欢

转载自blog.csdn.net/yanwucao/article/details/79872955