当函数期望数据帧行python时，使用apply()

Question

我的函数返回一个基于数据帧行的值

def function(df, row):
if df['A'][row]!='value':
return np.nan
else:
return df['B'][row]

我需要用apply来称呼它。但是，我不知道如何将行传递到函数中。我知道有更有效的方法可以做到这一点，但内部结构非常清楚(是的，这是一个类作业(

我尝试过apply()和lambda的多种变体。什么都不起作用。最近的尝试是这样的，但它期待着一个行参数。

df.apply(hourly_wage, axis=1)

有什么建议吗？

Answer 1

你试过了吗：df.apply(lambda row: function(row), axis=1)
但你必须更改你的功能：

def function(row):
if row['A']!='value':
return np.nan
else:
return row['B']