numpy中的meshgrid函数

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/zhangchuang601/article/details/79975544

语法:x,y = meshgrid(a,b)

作用:输入两个一维数组,输出两个二维数组x,y。其中矩阵x的行向量是向量a的简单复制,而矩阵y的列向量是向量b的简单复制。

示例:

a = np.linspace(0,2,3)
b = np.linspace(0,2,3)
c = zip(a.ravel(),b.ravel())#根据a,b的一维数据。打包为元组的列表
print(list(c))#注意在使用时要用list()函数处理,类似于map返回值的处理方法。
# [(0.0, 0.0), (1.0, 1.0), (2.0, 2.0)]

aa,bb = np.meshgrid(a,b)
print(aa)
# [[ 0.  1.  2.]
#  [ 0.  1.  2.]
#  [ 0.  1.  2.]]
print(bb)
# [[ 0.  0.  0.]
#  [ 1.  1.  1.]
#  [ 2.  2.  2.]]
xv,yv = np.meshgrid(a,b,sparse=True)  
print(xv)  
#[[ 0.  1.  2.]]  
print(yv)
# [[ 0.]
#  [ 1.]
#  [ 2.]]
xv.shape

阅读更多

没有更多推荐了,返回首页