Fipy 从网格区域而不是通过点坐标获取值

问题描述

我正在尝试从某个区域获取值。我知道有一种方法可以通过引入坐标来调用值：

var([(x1,x2,x3) #xvalues
     (y1,y2,y3)]) #yvalues

但是，例如，如果我之前使用 var.setValue（它接受区域）中的“where”属性设置值，那么它们有多少点和哪些坐标会令人困惑。

var.setValue(20.,where=(Y <= y1) & ((X <= x1) & (X > x2)))

有没有什么方法可以像 var.setValue(where='...') 一样通过“查询”一个区域来获取值？

谢谢。

解决方法

如果可以根据单元坐标为一个区域构建一个掩码，那么该掩码可用于提取值。 FiPy 中没有特殊函数来执行此操作，但我们可以使用布尔掩码来索引变量。例如，

from fipy import CellVariable,Grid2D

mesh = Grid2D(nx=4,ny=4)
var = CellVariable(mesh=mesh)
mask = (mesh.y <= 2) & (mesh.x <= 3) & (mesh.x > 1)
var.setValue(20.,where=mask)
print('x:',mesh.x[mask.value])
print('y:',mesh.y[mask.value])
print('val:',var[mask.value])
print(var.mesh)
print(var[mask.value].mesh)

给予

x: [1.5 2.5 1.5 2.5]
y: [0.5 0.5 1.5 1.5]
val: [20. 20. 20. 20.]
UniformGrid2D(dx=1.0,nx=4,dy=1.0,ny=4)
Traceback (most recent call last):
  File "tmp.py",line 11,in <module>
    print(var[mask.value].mesh)
AttributeError: 'unOp' object has no attribute 'mesh'

以上为您提供了值的子区域以及相应的坐标。然而，在 FiPy 中没有办法提取一个子区域变量，同时维护其关联的网格数据。