Python 如何为每个返回向量而不是单个值的组运行函数？

我想知道，我在熊猫里是怎么做的

我有一个数据集，在值列和组列上

   x group order
1 22     a     1
2 33     a     2
3 11     a     3
4  4     b     1
5 88     b     2
6 77     b     3
7 43     b     4
8  9     b     5

我想分析每组的值列。例如，我想使用fft。我如何在每个组上运行一个函数，返回一个序列，而不仅仅是一个值（对于每个组，fft生成一个向量），然后按行返回

我期待类似的事情

   y group  order
1 21     a      1
2 62     a      2
3 83     a      3
4  4     a      4
6 46     b      1
7 17     b      2

作为输出

---

我想用熊猫做这个。如果可以使用

完成，则使用应用并将结果包装在一个pd.Series中

df.groupby('group').x.apply(lambda x: pd.Series(np.random.choice(x, 2)))

group   
a      0    22
       1    33
b      0    88
       1    43
Name: x, dtype: int64

---

使用

apply

并将结果包装在

pd.Series

df.groupby('group').x.apply(lambda x: pd.Series(np.random.choice(x, 2)))

group   
a      0    22
       1    33
b      0    88
       1    43
Name: x, dtype: int64