3分钟理解np.meshgrid()[通俗易懂]

大家好，又见面了，我是你们的朋友全栈君。

官方解释: np.meshgrid(*xi, **kwargs) Return coordinate matrices from coordinate vectors. 从坐标向量中返回坐标矩阵 不够直观

直观的例子

二维坐标系中,X轴可以取三个值1,2,3, Y轴可以取三个值7,8, 请问可以获得多少个点的坐标? 显而易见是6个: (1,7)(2,7)(3,7) (1,8)(2,8)(3,8)

np.meshgrid()就是干这个的!

#coding:utf-8
import numpy as np
# 坐标向量
a = np.array([1,2,3])
# 坐标向量
b = np.array([7,8])
# 从坐标向量中返回坐标矩阵
# 返回list,有两个元素,第一个元素是X轴的取值,第二个元素是Y轴的取值
res = np.meshgrid(a,b)
#返回结果: [array([ [1,2,3] [1,2,3] ]), array([ [7,7,7] [8,8,8] ])]

同理还可以生成更高维度的坐标矩阵

发布者：全栈程序员栈长，转载请注明出处：https://javaforall.cn/130588.html原文链接：https://javaforall.cn