是否可以在 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)?