[SOLVED] DataFrame Creation from DataFrame.apply

DataFrame Creation from DataFrame.apply

I have a function that returns a pd.DataFrame given a row of another dataframe:

def func(row):
    if row['col1']== 1:
        return pd.DataFrame({'a':[1], 'b':[11]})
    else:
        return pd.DataFrame({'a':[-1, -2], 'b':[-11,-22]})

I want to use apply func to another dataframe to create a new data frame, like below:

df = pd.DataFrame({'col1':[1,2,3],'col2':[11,22,33]})

# do some cool pd.DataFrame.apply stuff
# resulting in the below dataframe
pd.DataFrame({
    'a':[1,-1,-2,-1,-2],
    'b':[11,-11,-22,-11,-22]
})

Currently, I use the code below for the desired result:

pd.concat([mini[1] for mini in df.apply(func,axis=1).iteritems()])

While this works, it is fairly ugly. Is there a more elegant way to create a dataframe from df?

Solution

You could use:

pd.concat(df.apply(func, axis=1).tolist())

Output: