/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

0 vote335 views1 answer
How to install pandas for aws sam local
Python
69 votes291 views6 answers
Python online compiler 3.8
Python
37 votes288 views4 answers
Python To find the sum of all the elements in a list.
Python
4 votes280 views1 answer
Python separate string in groups based on delimiter
Python
2 votes285 views10 answers
Javascript
Python
5 votes289 views6 answers
Convert int to binary python
Python
0 vote303 views1 answer
Tableau prep encoding to a set of fields in a flow
Python
0 vote290 views2 answers
Python discord embed link
Python
1 vote285 views1 answer
Python wikipedia api search
Python
0 vote317 views1 answer
Pendulum python
Python