为给定三个顶点的等高线图生成三角形遮罩

2024-09-22 16:28:21 发布

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

以下是我生成的一些随机数据:

import scipy.signal as sgn
import scipy.interpolate as intr
import numpy.ma as ma
x = np.linspace(-100,0,500)
y = sgn.sawtooth(2 * np.pi * .2 * x)

y = (sgn.sawtooth(2 * np.pi * .2 * x)+1)/2
y = (y+1)*25
plt.plot(x,y)
z = np.sin(2*np.pi*.1*x)+np.sin(2*np.pi*.1*y)

这给了我一个数字: enter image description here

然后我建立一个等高线图:

xi,yi = np.meshgrid(np.linspace(x.min(),x.max(),200),np.linspace(y.min(),y.max(),200))   
zi = intr.griddata((x,y), z, (xi, yi) , method='cubic')    
plt.contourf( xi,yi,zi,100); plt.colorbar()

enter image description here s码

对于方形面具,我会这样做:

xi,yi = np.meshgrid(np.linspace(x.min(),x.max(),200),np.linspace(y.min(),y.max(),200))
mask =(yi> 25) & (yi< 35) & (xi > -55) & (xi < -25) 
zi = intr.griddata((x,y), z, (xi, yi) , method='cubic')
zi = ma.masked_array(zi, mask = mask )
plt.contourf( xi,yi,zi,100); plt.colorbar() 

enter image description here

我的问题是,在给定三角形的三个坐标(-50,0),(-25,0),(-37,25)的情况下,如何把一个三角形掩模放在与正方形相同的基位置上。你知道吗


Tags: importasnppipltmaskminmax
1条回答
网友
1楼 · 发布于 2024-09-22 16:28:21

试着按照这些思路来做:

mask =(yi> 25) & (yi< 35) & (xi > -55) & (xi < -25) & ((xi+95) > 2*yi)

换句话说,把((xi+95) > 2*yi)加到你的面具上。你可能需要调整一些常数来得到你想要的。你知道吗

相关问题 更多 >