numpy的生成网格矩阵 meshgrid()

Posted 2020-12-27 gengyi

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 维矩阵

示例2 

交换两参数的顺序

import numpy as np

a = [1,2,3]
b = [9,8,7]

c, d = np.meshgrid(b,a)

print(c)
# [[9 8 7]
#  [9 8 7]
#  [9 8 7]]
print(d)
# [[1 1 1]
#  [2 2 2]
#  [3 3 3]]

交换两个参数顺序后，输出结果发生了变化。

示例3

当返回值值是两个或两个以上参数时，也可用一个参数来接受。

import numpy as np
a = [1,2,3]
b = [9,8,7]
c = np.meshgrid(a,b)
print(c)
# 下面是打印出的结果+
# [array([[1, 2, 3],
#        [1, 2, 3],
#        [1, 2, 3]]), array([[9, 9, 9],
#        [8, 8, 8],
#        [7, 7, 7]])]

2.2.2 两个参数长度不一致时