python中如何实现复杂函数

在python中利用表格型创建一个复杂函数，具体方法如下：

import math
df = pd.DataFrame({"code" : [1,1,2,2,3], "value" : [10, 11, 11, 11, 30]})
cc = df.groupby(['code','value']).size().reset_index().rename(columns={0:'count'})
cc = cc.assign(result=cc.apply(lambda r: math.sqrt(r.code**2 + r.value**3), axis=1)).rename(columns={"code":"col1","value":"col2"}).drop(columns="count")