numpy meshgrid函数

函数原型

numpy.meshgrid(*xi, copy=True, sparse=False, indexing='xy')

函数说明

meshgrid函数用于从坐标向量中返回坐标矩阵。
比如，x向量为[1, 2, 3]，y向量为[4, 5]，则由x向量中的点和y向量中点可能的组合为：
(1, 4), (2, 4), (3, 4)
(1, 5), (2, 5), (3, 5)
对应于坐标矩阵为：
[[1, 2, 3],
[1, 2, 3]]
[[4, 4, 4],
[5, 5, 5]]
第一个矩阵对应于所有组合中的x点，第二个矩阵对应于y点。那么meshgrid函数的作用就是根据这两个向量x、y生成对应的坐标矩阵。

函数使用

>>> a = np.array([1, 2, 3])
>>> a
array([1, 2, 3])
>>> b = np.array([4, 5])
>>> b
array([4, 5])
>>> c = np.meshgrid(a, b)
>>> c
[array([[1, 2, 3],
       [1, 2, 3]]), 
 array([[4, 4, 4],
       [5, 5, 5]])]