python笔记:numpy中mgrid的用法

时间:2024-05-20 13:36:49
np.mgrid[start:end:step]
参数解析:

    start:开始坐标

    stop:结束坐标(实数不包括,复数包括)

    step:步长

用法:返回多维结构,常见的如2D图形,3D图形。

第1返回值为第1维数据在最终结构中的分布,

第2返回值为第2维数据在最终结构中的分布,以此类推。(分布以矩阵形式呈现)  

mgrid[[1:3:3j, 4:5:2j]]  
  • 3j:3个点
  • 步长为复数表示点数左闭右闭
  • 步长为实数表示间隔左闭右开
  • 例如1D结构(array),如下:

python笔记:numpy中mgrid的用法

  • 例如2D结构 (2D矩阵),如下: 

python笔记:numpy中mgrid的用法

python笔记:numpy中mgrid的用法

  •  例如3D结构 (3D立方体),如下: 
In [13]: import numpy as np

In [14]: x,y,z=np.mgrid[-5:5:3j,-2:2:3j,-1:1:2j]
In [15]: x
Out[15]: 
array([[[-5., -5.],
        [-5., -5.],
        [-5., -5.]],

       [[ 0.,  0.],
        [ 0.,  0.],
        [ 0.,  0.]],

       [[ 5.,  5.],
        [ 5.,  5.],
        [ 5.,  5.]]])
In [16]: y
Out[16]: 
array([[[-2., -2.],
        [ 0.,  0.],
        [ 2.,  2.]],

       [[-2., -2.],
        [ 0.,  0.],
        [ 2.,  2.]],

       [[-2., -2.],
        [ 0.,  0.],
        [ 2.,  2.]]])
In [17]: z
Out[17]: 
array([[[-1.,  1.],
        [-1.,  1.],
        [-1.,  1.]],

       [[-1.,  1.],
        [-1.,  1.],
        [-1.,  1.]],

       [[-1.,  1.],
        [-1.,  1.],
        [-1.,  1.]]])