numpy模块中的meshgrid函数用来生成网格矩阵,最简单的网格矩阵为二维矩阵
meshgrid函数可以接受 x1, x2,..., xn 等 n 个一维向量,生成 N-D 矩阵。
1 基本语法
meshgrid(*xi, **kwargs)
参数:
xi - x1, x2,..., xn : array_like
返回值:
X1, X2,..., XN : ndarray
2 示例(二维网格)
2.1 一个参数时
import numpy as np
a = [1,2,3]
b = np.meshgrid(a)
print(b) # [array([1, 2, 3])]
当只有一个参数时,返回值也只有一个 b ,若写两个返回值 b, c = np.meshgrid(a) 则会报错。
2.2 两个参数时
2.2.1 两个参数长度一致时
示例1
import numpy as np
a = [1,2,3]
b = [9,8,7]
c, d = np.meshgrid(a,b)
print(c)
print('-'*10)
print(d)
运行
[[1 2 3]
[1 2 3]
[1 2 3]]
----------
[[9 9 9]
[8 8 8]
[7 7 7]]
当两个参数长度一致时(如长度为 N ),则生成 N * N 维矩阵