python - 如何使用列表将用户定义的函数更改为pandas系列

Question

我看到一篇关于创建递归求和的帖子 recursive cumulative sums 我对其进行了修改以添加衰减“因子”

def rec(n, factor):
    if len(n) < 2: return n
    n[1] = factor*n[0] + n[1]
    return [n[0]] + rec(n[1:], factor)

print(rec([1,2,3,4], 0.5))
#[1, 2.5, 4.25, 6.125]

该函数适用于列表。我想将其更改为用于 pandas 系列。因此，我可以使用“transform”将每组“ID”应用于 DataFrame。如下所示:

import pandas as pd

df_dic = {'ID': [1,1,1,1,2,2,2],
         'channel': [1,2,3,4,3,2,4]}
df = pd.DataFrame.from_dict(df_dic)

output = df.groupby(['ID'])['channel'].transform(rec, 0.5)

为了做到这一点，我想我需要修改该函数以使用 pandas 系列而不是列表。有没有简单的方法可以做到这一点？

Answer 1

在这种情况下，您可以将系列转换为列表，如下所示:

def rec(n, factor):
    if (type(n) != list): n = list(n)  # <---- here it is
    if len(n) < 2: return n
    n[1] = factor*n[0] + n[1]
    return [n[0]] + rec(n[1:], factor)

print(rec([1,2,3,4], 0.5))
#[1, 2.5, 4.25, 6.125]


import pandas as pd

df_dic = {'ID': [1,1,1,1,2,2,2],
     'channel': [1,2,3,4,3,2,4]}
df = pd.DataFrame.from_dict(df_dic)

output = df.groupby(['ID'])['channel'].transform(rec, 0.5)


Out[1]:
    0    1.000
    1    2.500
    2    4.250
    3    6.125
    4    3.000
    5    3.500
    6    5.750