pandas explode column only non zero values

user_1616785 0
Viewed 60

How to filter tolist or values below to get only non-zeros

import pandas as pd
df = pd.DataFrame({'A':[0,2,3],'B':[2,0,4], 'C': [3,4,0]})
df['D']=df[['A','B','C']].values.tolist()
df.explode('D')

Data

    A   B   C
0   0   2   3
1   2   0   4
2   3   4   0

On Explode on Column D rows now becomes 9. But i want 4 rows in the output

Expected result

    A   B   C   D
0   0   2   3   2
0   0   2   3   3
1   2   0   4   2
1   2   0   4   4 
2   3   4   0   3
2   3   4   0   4

I got list(filter(None, [1,0,2,3,0])) to return only non-zeros. But not sure how to apply it in the above code

pandas python
2 Answers

index.repeat

m = df.ne(0)
df.loc[df.index.repeat(m.sum(1))].assign(D=df.values[m])

   A  B  C  D
0  0  2  3  2
0  0  2  3  3
1  2  0  4  2
1  2  0  4  4
2  3  4  0  3
2  3  4  0  4
user_128331660

Simpliest is query:

df['D']=df[['A','B','C']].values.tolist()
df.explode('D').query('D != 0')

Output:

   A  B  C  D
0  0  2  3  2
0  0  2  3  3
1  2  0  4  2
1  2  0  4  4
2  3  4  0  3
2  3  4  0  4
user_42384080
Related