meshgrid
meshgrid的作用是:接受两个一维数组生成两个二维矩阵,对应两个数组中所有的(x,y)对, 生成网络
import numpy as np
x = [1, 2, 3]
y = [4, 5, 6, 7]
x1, y1 = np.meshgrid(x, y)
print(x1)
print(y1)
z = np.meshgrid(x, y)
print(z)
print(np.array(x1).shape, np.array(y1).shape)
其实一开始我并不懂x1和y1为什么是这样的,也不懂为什么说“矩阵对应两个数组中所有的(x,y)对”,从图中可以看到x1和y1都是4*3的矩阵,我们将对应位置的元素放在一起,可以得到:
for i in range(0, 4):
for j in range(0, 3):
print("(", x1[i, j], ",", y1[i, j], ")")
( 1 , 4 )
( 2 , 4 )
( 3 , 4 )
( 1 , 5 )
( 2 , 5 )
( 3 , 5 )
( 1 , 6 )
( 2 , 6 )
( 3 , 6 )
( 1 , 7 )
( 2 , 7 )
( 3 , 7 )
这样看,就可以看出二维矩阵对应的所有的(x,y)对.
生成的数组的维度是y的维度*x的维度。
x = np.zeros((3, 5))
y = np.zeros((4, 2))
x1, y1 = np.meshgrid