Python函数不起作用

时间:2017-02-16 11:47:14

标签: python function pandas

我有以下df:

df = pd.DataFrame(columns=['mbs','Wholesale Data Usage'], index=['x','y','z'])
df.loc['x'] = pd.Series({'mbs':32, 'Wholesale Data Usage':36})
df.loc['y'] = pd.Series({'mbs':64, 'Wholesale Data Usage':62})
df.loc['z'] = pd.Series({'mbs':256, 'Wholesale Data Usage':277})

此外我定义了以下功能:

def calculate_costs(row):
    mbs = row.loc['mbs']
    wdu = row.loc['Wholesale Data Usage']

    ac = 0
    if wdu >= mbs:
        if mbs == 32 | mbs == 64:
            ac = (wdu - mbs) * 0.05
        elif mbs == 128 | mbs == 256:
            ac = (wdu - mbs) * 0.036
        elif mbs == 512 | mbs == 1024:
            ac = (wdu - mbs) * 0.018

    return ac

由于某些原因,如果我将函数应用于我的df,则所有行值均为0:

df['Additional Charge'] = df.apply(lambda r: calculate_costs(r), axis=1)

你能告诉我我做错了吗?

1 个答案:

答案 0 :(得分:1)

我不确定这是否是你想要的,结果现在不是零。 (我将|符号更改为“或”)

def calculate_costs(row):
    mbs = row.loc['mbs']
    wdu = row.loc['Wholesale Data Usage']
    print(mbs,wdu)
    ac = 0
    if wdu >= mbs:
        print("Hej")
        if mbs == 32 or mbs == 64:
            ac = (wdu - mbs) * 0.05
        elif mbs == 128 or mbs == 256:
            ac = (wdu - mbs) * 0.036
        elif mbs == 512 or mbs == 1024:
            ac = (wdu - mbs) * 0.018
        else:
            print ("Hello")
    return ac