使用 DataFrame.assign() 为多个列赋值

Question

我有一个存储在 pandas 数据框df 中的字符串列表，列名是文本（即df['text']）。 我有一个函数f(text: str) -> (int, int, int)。 现在，我想做以下事情。

df['a'], df['b'], df['c'] = df['text'].apply(f)

如何使用函数的三个返回值创建三列？

上面的代码给出了错误

ValueError: too many values to unpack (expected 3)

我试过了

df['a', 'b', 'c'] = df['text'].apply(f)

但我得到一列名称为'a', 'b', 'c'

注意：

1. SO 中有一个similar question，但是当我从那里使用以下解决方案时，我再次遇到错误。

   df[['a', 'b', 'c']] = df['text'].apply(f, axis=1, result_type='expand')
   

   错误是

   f() got an unexpected keyword argument 'axis'
   f() got an unexpected keyword argument 'result_type' #(once I remove the axis=1 parameter)
   

   2. 注意df还有其他列

Answer 1

对我来说，您的解决方案有效。但是需要测试元组中是否正确返回 3 个值。

这是替代方案：

df = pd.DataFrame({'text':[1,2,3]})

def f(x):
    return((x,x+1,x-5))

df[['a', 'b', 'c']] = pd.DataFrame(df['text'].apply(f).tolist(), index=df.index)
print (df)
   text  a  b  c
0     1  1  2 -4
1     2  2  3 -3
2     3  3  4 -2