熊猫在多列上应用元组解包功能

时间:2018-10-18 14:43:56

标签: python-3.x pandas function tuples apply

给出一个function,它需要多个参数并按如下方式返回多个值:

def tuple_unpack(value, another_value):
  ''' does some interesting stuff ... '''
   return value, another_value

有没有一种方法可以将这样的函数应用于pandas dataframe,其中对于2个函数参数,我可以从2列中传递值,然后将输出元组解包为多个列: >

df[['value_col','another_value_col']] = df.apply(lambda df.col, df.col: tuple_unpack)  

1 个答案:

答案 0 :(得分:1)

您可以将concat与数据框构造函数一起使用

unpackdf=pd.DataFrame(df.apply(lambda x : tuple_unpack(x.col_1,x.col_2),1).tolist(),columns=['col1','col2'],index=df.index)
yourdf=pd.concat([unpackdf,df],axis=1)