在Python中对x,y网格执行计算

Question

我想对X, Y数据执行计算以产生计算的z。我的代码如下:

injection_wells.csv示例数据集

<表类> 名称 X Y Q tbody><<tr>MW-125177009964005MW-225177709964205

在看了更多的内容之后，我认为我现在已经理解了这个问题，所以我要在这里过早地进行讨论。

所以你得到了一个注水井数据的DataFrame，有四列:

name    x    y    q
str   int  int  int

你有一个想求值的函数f(x, y, q) -> z。我不确定我是否完全遵循你的函数在做什么，因为它的格式很难阅读，所以我将使用一个简化的例子:

def func(x, y, q):
return (q / 2 * np.pi) * np.arctan(y, x)

现在不需要将井数据分解为不同的数组，只需在整个数据框架上逐行应用该函数:

df["z"] = func(df.x, df.y, df.q)