如何在每个返回向量的组中运行函数,而不是使用单个值?

2024-03-28 16:23:21 发布

您现在位置: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

作为输出。你知道吗

我想在熊猫身上做这个。如果可以用https://github.com/kieferk/dfply来做的话,就要加分


Tags: 数据函数httpsfftgithubcomgrouporder
1条回答
网友
1楼 · 发布于 2024-03-28 16:23:21

使用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

相关问题 更多 >