I have a pandas DataFrame where one column contains lists and would like to select the rows where the lists are not empty.
Example data:
df = pd.DataFrame({'letter': ["a", "b", "c", "d", "e"],
'my_list':[[0,1,2],[1,2],[],[],[0,1]]})
df
letter my_list
0 a [0, 1, 2]
1 b [1, 2]
2 c []
3 d []
4 e [0, 1]
What I'd like:
df
letter my_list
0 a [0, 1, 2]
1 b [1, 2]
4 e [0, 1]
What I'm trying:
df[df.my_list.map(lambda x: if len(x) !=0)]
... which returns an invalid syntax error. Any suggestions?
Or you can follow your own logic using length of the list to determine keep or not .
df[df.my_list.str.len().gt(0)]
Out[1707]:
letter my_list
0 a [0, 1, 2]
1 b [1, 2]
4 e [0, 1]
Empty lists evaluate to False
in a boolean context
df[df.my_list.astype(bool)]
letter my_list
0 a [0, 1, 2]
1 b [1, 2]
4 e [0, 1]
If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!
Donate Us With