python - Python 和 Pandas 的问题 : Adding calculated column to dataframe that includes data from a function provides error

Question

我正在尝试将计算列:“NetEarnings”添加到我的 DataFram“Wages”中。 “净利润”栏从年收入中减去税款；税是从我创建的用于计算税的函数中提取的。由于错误，它不会让我添加这个新列:

"TypeError: unsupported operand type(s) for -: 'float' and 'tuple'"

我已经尝试了几乎所有的方法，不确定我是否在某个地方犯了一个愚蠢的错误。感谢您的帮助!

代码:

def Tax(AnnualIncome):
    if (0 < AnnualIncome) & (AnnualIncome <= 21450):
        return (.15 * AnnualIncome)
    elif (21450 < AnnualIncome) & (AnnualIncome <= 51900):
        return (3215.5 + ((AnnualIncome - 21450) * .28))
    else: 
        return (11,743.5 + ((AnnualIncome - 51900) * .31))

wages['Tax'] = wages['AnnualIncome'].apply(Tax)

# Problem line
wages['NetEarning'] = wages['AnnualIncome'] - wages['Tax']

错误:

TypeError: unsupported operand type(s) for -: 'float' and 'tuple'

Answer 1

除了 Ente 答案之外，

我建议使用 np.where 而不是 apply。 apply 比 for 循环快，但比 apply 慢得多。

一个可能的解决方案是:

np.where(df['AnnualIncome'] <= 21450, (.15 *df['AnnualIncome']), 
               np.where(df['AnnualIncome'] <= 51900,
                        (3215.5 + ((df['income']- 21450) * .28)),
                              (11743.5 + (df['AnnualIncome'] - 51900) * .31)
                ) 
        )