/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

48 votes133 views6 answers
Python to exe
Python
0 vote136 views1 answer
Line continutation in r string python
Python
21 votes136 views6 answers
Convert datetime to date pandas
Python
65 votes124 views6 answers
Spanish dictionary
Python
46 votes130 views6 answers
Round to two decimal places python
Python
16 votes144 views4 answers
Random.choice
Python
0 vote126 views1 answer
Pyplot aera
Python
8 votes143 views3 answers
Export multiple python pandas dataframe to single excel file
Python
0 vote266 views1 answer
Pandas index from 1
Python
15 votes150 views6 answers
Infinity python
Python