Remove nan from list if some of the elements are data-frame

Brian Smith 0
Viewed 29

I have below list

import pandas as pd
import numpy as np
dat = pd.DataFrame({'name' : ['A', 'C', 'A', 'B', 'C'], 'val' : [1,2,1,2,4]})
List = [dat, np.nan, dat, np.nan, np.nan]
List

I want to retain only those elements where they are not nan.

There is a similar discussion in How can I remove Nan from list Python/NumPy, but it is only applicable if all elements are scalar.

Is there any way to remove nan if some elements are matrix/dataframe?

pandas numpy python-3.x
1 Answers

You can do:

[x for x in List if isinstance(x, pd.DataFrame)]

If you insist on filtering on filter out nan only, then remember that it is a float instance, so:

[x for x in List if not(isinstance(x, float) and np.isnan(x))]
Quang Hoang0
Related