我想对X, Y数据执行计算以产生计算的z。我的代码如下:
injection_wells.csv示例数据集
<表类>
名称
X
Y
Q
tbody><<tr>MW-1 2517700 996400 5 MW-2 2517770 996420 5 表类>
在看了更多的内容之后,我认为我现在已经理解了这个问题,所以我要在这里过早地进行讨论。
所以你得到了一个注水井数据的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)