python - 如何在 Python 中创建复杂的函数？

Question

我有如下 Python 数据框：

df = pd.DataFrame({"code" : [1,1,2,2,3], "value" : [10, 11, 11, 11, 30]})

接下来，我发现没有双重的组合代码 x 值

cc = df.groupby(['code','value']).size().reset_index().rename(columns={0:'count'})
cc

接下来，我需要创建函数，其中将使用来自 cc 的代码 x 值的每个组合，然后保存在 Data Frame 中，例如

对于索引 0 是代码 = 1 和值 = 10，我需要函数将迭代每个索引的组合代码 x 值并总结这些值，例如：
对于索引 0 -> 1+10 = 11
对于索引 1 - > 1+11 = 12
用于索引 2 ->2+11 = 13
用于索引 3 -> 3 +30 = 33
并在数据框中返回函数结果，如下所示：

Answer 1

这是一个非常简单的功能......你想要应用的任何功能

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")