/Python/Expand pandas dataframe into separate rows

Expand pandas dataframe into separate rows

def explode(df, lst_cols, fill_value='', preserve_index=False):
    # make sure `lst_cols` is list-alike
    if (lst_cols is not None
        and len(lst_cols) > 0
        and not isinstance(lst_cols, (list, tuple, np.ndarray, pd.Series))):
        lst_cols = [lst_cols]
    # all columns except `lst_cols`
    idx_cols = df.columns.difference(lst_cols)
    # calculate lengths of lists
    lens = df[lst_cols[0]].str.len()
    # preserve original index values    
    idx = np.repeat(df.index.values, lens)
    # create "exploded" DF
    res = (pd.DataFrame({
                col:np.repeat(df[col].values, lens)
                for col in idx_cols},
                index=idx)
             .assign(**{col:np.concatenate(df.loc[lens>0, col].values)
                            for col in lst_cols}))
    # append those rows that have empty lists
    if (lens == 0).any():
        # at least one list in cells is empty
        res = (res.append(df.loc[lens==0, idx_cols], sort=False)
                  .fillna(fill_value))
    # revert the original index order
    res = res.sort_index()
    # reset index if requested
    if not preserve_index:        
        res = res.reset_index(drop=True)
    return res

Source: stackoverflow.com

Similar codes

-2 votes267 views1 answer
What is xbox live called now
Python
73 votes263 views6 answers
Reverse the string in python
Python
0 vote272 views1 answer
Python convert dict to xml
Python
1 vote278 views1 answer
How to write words on any other apps in python
Python
90 votes296 views6 answers
Python get current directory
Python
14 votes295 views1 answer
Create data dir in python
Python
0 vote270 views1 answer
Comparison of two csv file and output with differences?
Python
0 vote285 views1 answer
Pandas, python, dataframes
Python
1 vote278 views1 answer
Show documentation or information about a function/ method in jupyter notebook
Python
4 votes247 views1 answer
Declaring a numpy array with certain size
Python