python - pandas Dataframe分配问题

Question

我想使用 .assign 函数添加新列。

df = pd.DataFrame({'A': range(1, 5), 'B': range(11, 15)})

def delta(df):

    df = df.assign(df_delta = df.A - df.B,
                   df_multiply = df_delta*30
                   )

    return df


print(delta(df))

但出现此错误

NameError: name 'df_delta' is not defined

Answer 1

或者您可以使用 lambda 来实现此目的，它可以读取已分配的列:

def delta(df):
    df = df.assign(df_delta = df.A - df.B, df_multiply = lambda x : x['df_delta']*30)
    return df

print(delta(df))

<小时/>

   A   B  df_delta  df_multiply
0  1  11       -10         -300
1  2  12       -10         -300
2  3  13       -10         -300
3  4  14       -10         -300