3

是否可以在 groupby 之后将用户函数“应用”到 python 数据表?

例如:

import datatable as dt
from datatable import f, by, sum
df = dt.Frame(SYM=['A','A','A','B','B'], xval=[1.1,1.2,2.3,2.4,2.5])
print(df[:, sum(f.xval), by(f.SYM)])

这行得通。但我想用使用以下定义的用户函数替换“sum”函数:

def func(x):
    # do some operations here; e.g. ranking
    y = x
    return(y)    

这可能吗?您能否提供一个示例(可能在上面的 func 中使用 numpy.rank)?

4

0 回答 0