Pandas:将一个包含列和一个变量的函数应用为argumen

2024-04-25 06:43:51 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图将一个函数应用于一个dataframe,它有多个参数,其中两个需要分配给dataframe的行,一个是变量(一个简单的数字)。在

类似线程的一个变体适用于行:(与我原来的函数相比,所有函数都过于简化)

import pandas as pd

dict={'a':[-2,5,4,-6], 'b':[4,4,5,-8]}

df=pd.DataFrame (dict)
print(df)

def DummyFunction (row):
    return row['a']*row['b']
#this works:
df['Dummy1']=df.apply(DummyFunction, axis=1)

但是我如何应用下面的变量,我的函数接受一个额外的参数(一个固定变量)?我似乎无法在apply方法中传递它:

^{pr2}$

谢谢你的帮助!在


Tags: 函数importdataframepandasdf参数as数字
1条回答
网友
1楼 · 发布于 2024-04-25 06:43:51

可以将附加变量作为命名参数直接传递给pd.DataFrame.apply

def DummyFunction2(row, threshold):
    return row['a']*row['b']*threshold

df['Dummy2'] = df.apply(DummyFunction2, threshold=2, axis=1)

相关问题 更多 >