现在我有一个数据框:
df = pd.DataFrame({"a":[1,2,3,4,5],"b":[2,3,4,5,6],"c":[3,4,5,6,7]})
功能:
def fun(a,b,shift_len):
return a+b*shift_len,b-shift_len
Run Code Online (Sandbox Code Playgroud)
我可以通过以下方式得到结果:
df[["d","e"]] = df.apply(lambda row:fun(row["a"],row["b"],3),axis=1,result_type="expand")
Run Code Online (Sandbox Code Playgroud)
我想知道如何使用极坐标来获得相同的结果?