熊猫按行组应用卷积

Question

>我有一个熊猫数据帧，如下所示：

profile_index  point_index         z              x             y
0              0            1 -0.885429  297903.323027  6.669492e+06
1              0            2 -0.820151  297904.117752  6.669492e+06
2              0            3 -0.729671  297904.912476  6.669491e+06
3              0            4 -0.649332  297905.707201  6.669490e+06
4              1            1 -0.692186  297906.501926  6.669490e+06
5              1            2 -0.885429  297903.323027  6.669492e+06
6              1            3 -0.820151  297904.117752  6.669492e+06
3              1            4 -0.649332  297905.707201  6.669490e+06

我想通过在向量(Z列(上应用卷积(numpy.convolve(来创建一个新的"z_gauss"列，该向量(列z(对应于数据帧中具有相同"profile_index"的一组行。

我尝试做类似的事情

data["z_gauss"] = data.groupby('profile_index').apply(lambda x: np.convolve(x, gaussian, 'same'))

其中gaussian是我的高斯滤波器(矢量(。但是我得到一些错误，例如ValueError: object too deep for desired array

您对如何进行有任何建议/提示吗？我应该将数据帧拆分为不同的数据帧吗？

Answer 1

您希望使用转换而不是应用。这将避免插入每行组大小的向量：

data["z_gauss"] = (data.groupby('profile_index')['z']
.transform(lambda x: np.convolve(x, gaussian, 'same')))