如何在一个绘图中结合python中的曲面和箭袋绘图

2024-05-15 15:15:23 发布

您现在位置:Python中文网/ 问答频道 /正文

我有下面的代码,它生成一个曲面图,曲面作为网格X,Y的z坐标

该代码还生成一个向量场图,作为相对于同一坐标系X,Y的单独图形

我有没有办法把这两个数字合并成一个数字?向量场是在曲面上还是在曲面图的底部

import numpy as np
import matplotlib.pyplot as plt
import grad_field
 # first create the plot of the elevation surface
fig, ax = plt.subplots(subplot_kw={"projection": "3d"})
LNG = np.linspace(lngmin,lngmax,samples1)
LAT = np.linspace(latmin,latmax, samples2)
X, Y = np.meshgrid(LNG,LAT)
ax.plot_surface(X, Y, elev_mat)
plt.show()
 
# now create the vector field plot of the gradients
[gradx,grady] = grad_field.grad_field(elev_mat)
fig1, ax1 = plt.subplots()
ax1.set_title('Arrows scale with plot width, not view')
ax1.quiver(X, Y, gradx, grady, units='xy' ,scale=2, color='red')
plt.show()

Tags: ofthe代码importfieldplotascreate