将取决于数据框的索引和列的函数应用于数据框

Question

我有一个数据框df，其索引为[x[0], ..., x[N]]，列为[y[0], ..., y[M]]，其数据为z[i,j]的二维数组。

我有一个包含 3 个浮点变量的 python 函数 def f(x, y, z)，我想使用 numpy 和/或 pandas 以最快的方式计算 f(x[i], y[j], z[i,j]) 的二维数组，但我不知道该怎么做。

我看到了 df.transform 方法，但它似乎不允许依赖于 df 的索引和列的 lambda——或者至少我不知道如何提供这样的 lambda。

Answer 1

你可以做一个pd.melt：

df.reset_index().rename(columns={'index':'x'}).melt(var_name='y', value_name='z', id_vars='x')