python [Pandas]从多列创建多个列

Posted 2021-05-09

df = pd.DataFrame({
  "a": [1, 2 ,3],
  "b": [4, 5, 6],
  "c": [7, 8, 9],
})

def example(x):
  x['d'] = x['a'] ** 2
  x['e'] = x['b'] ** 3
  x['f'] = x['c'] ** 4
  return x
  
new_df = df.apply(example, axis = 1)