带循环功能或跨类别应用插值的lambda

时间:2019-06-24 22:43:24

标签: python pandas scipy

我试图使用多个3d曲面可视化数据集,并且必须先对数据进行平滑处理,为此我决定使用scipy插值函数Rbf。

我有很多类别,希望能够同时为每个或多个(或全部)绘制3d曲面。我打算使用可绘制的Mesh3d图。

我尝试了功能,并且到目前为止尝试使用lambda都没有成功。我怀疑循环也可以解决,但被嵌套弄糊涂了。希望能得到一个提示。

d = {'Id': ['A1', 'A2', 'A3', 'A4', 'A5', 'A6'], 'Category': ['Adam', 'Eve', 'Cain', 'Adam', 'Eve', 'Adam'],'X': [10, 20, 30, 40, 50, 60], 'Y': [0.1, 0.6, 0.75, 0.1, 0.9, 0.8], 'Z': [17, 13, 23, 29, 37, 43]}
df = pd.DataFrame(data=d)

def interpolation(x, y, z, sm):
    f = Rbf(x, y, z, smooth=sm)
    return f(x, y)

df[Z_smoothed] = df.groupby('Category').apply(interpolation(df.X, df.Y, df.Z, 3)

0 个答案:

没有答案