python pandas函数返回多个值并将它们放入列中

Question

我有两个功能如下。第二个有效，但第一个无效。为什么？

import pandas as pd

df = pd.DataFrame(data = {'a': [1, 2, 3], 'b': [4, 5, 6]})

def add_subtract_list(a, b):
  return [1,2,3,4]
add_subtract_list(5,8)

df[['c','d','x','y']] = df.apply(
    lambda row: add_subtract_list(row['a'], row['b']), axis=1)

以下功能有效。来自this页面

import pandas as pd

df = pd.DataFrame(data = {'a': [1, 2, 3], 'b': [4, 5, 6]})



def add_subtract_list(a, b):
  return [a + b, a - b]

df[['sum', 'difference']] = df.apply(
    lambda row: add_subtract_list(row['a'], row['b']), axis=1)

Answer 1

更新您的函数，使其返回熊猫系列而不是 python 列表。

def add_subtract_list(a, b):
  return pd.Series([1,2,3,4])