我有一个名为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()
这对我来说不起作用,因为它修改了原始数据框。一个潜在的解决方案是使用join,但这不允许我命名它,也不允许它被填充标量值:
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()
modified = original.copy(); modified['c'] = ...
- Viktor Kerkez