python - 将列作为副本添加到 Pandas DataFrame

Question

我有一个名为 original 的 pandas DataFrame，我想向其添加一个新列并将生成的 DataFrame 保存在一个名为 modified 的变量中。我怎么做？

import pandas as pd
import numpy as np
original = pd.DataFrame(np.random.randn(5, 2), columns=['a', 'b'])

此处名称非常相似的问题中给出的解决方案是执行以下操作:

original['c'] = original['b'].abs()

这对我不起作用，因为它修改了原始 DataFrame。一个潜在的解决方案是使用连接，但这不允许我命名它，也不允许它用标量值填充:

modified = original.join(original['b'].abs(),rsuffix='_abs')

目的是能够在没有临时变量的情况下在一行中添加列以实现以下效果:

modified = original.some_op() \
    .a_different_op() \
    .add_a_column() \ # <- the step I can't figure out
    .another_op() \
    .final_op()

Answer 1

使用 pandas.DataFrame.assign 方法在这里描述 http://pandas.pydata.org/pandas-docs/stable/generated/pandas.DataFrame.assign.html