Cómo encontrar qué columnas contienen cualquier valor NaN en Pandas dataframe (python)


Dado un dataframe pandas que contiene (tal vez) valores NaN dispersos aquí y allá:

Pregunta: ¿cómo puedo determinar qué columnas contienen valores NaN? En particular, ¿puedo obtener una lista de los nombres de columna que contienen NaNs?

Gracias

Author: MaxU, 2016-03-25

2 answers

ACTUALIZACIÓN: usando Pandas 0.22.0

Las nuevas versiones de Pandas tienen nuevos métodos 'DataFrame.isna ()' and ' DataFrame.notna()'

In [71]: df
Out[71]:
     a    b  c
0  NaN  7.0  0
1  0.0  NaN  4
2  2.0  NaN  4
3  1.0  7.0  0
4  1.0  3.0  9
5  7.0  4.0  9
6  2.0  6.0  9
7  9.0  6.0  4
8  3.0  0.0  9
9  9.0  0.0  1

In [72]: df.isna().any()
Out[72]:
a     True
b     True
c    False
dtype: bool

Como lista de columnas:

In [74]: df.columns[df.isna().any()].tolist()
Out[74]: ['a', 'b']

Para seleccionar esas columnas (que contienen al menos un valor NaN):

In [73]: df.loc[:, df.isna().any()]
Out[73]:
     a    b
0  NaN  7.0
1  0.0  NaN
2  2.0  NaN
3  1.0  7.0
4  1.0  3.0
5  7.0  4.0
6  2.0  6.0
7  9.0  6.0
8  3.0  0.0
9  9.0  0.0

Respuesta antigua:

Intenta usar isnull():

In [97]: df
Out[97]:
     a    b  c
0  NaN  7.0  0
1  0.0  NaN  4
2  2.0  NaN  4
3  1.0  7.0  0
4  1.0  3.0  9
5  7.0  4.0  9
6  2.0  6.0  9
7  9.0  6.0  4
8  3.0  0.0  9
9  9.0  0.0  1

In [98]: pd.isnull(df).sum() > 0
Out[98]:
a     True
b     True
c    False
dtype: bool

O como @root propuso la versión más clara:

In [5]: df.isnull().any()
Out[5]:
a     True
b     True
c    False
dtype: bool

In [7]: df.columns[df.isnull().any()].tolist()
Out[7]: ['a', 'b']

Para seleccionar un subconjunto-todas las columnas que contienen al menos un valor NaN:

In [31]: df.loc[:, df.isnull().any()]
Out[31]:
     a    b
0  NaN  7.0
1  0.0  NaN
2  2.0  NaN
3  1.0  7.0
4  1.0  3.0
5  7.0  4.0
6  2.0  6.0
7  9.0  6.0
8  3.0  0.0
9  9.0  0.0
 111
Author: MaxU,
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/ajaxhispano.com/template/agent.layouts/content.php on line 61
2018-01-18 15:30:33

Puede usar df.isnull().sum(). Muestra todas las columnas y el total de NANS de cada característica.

 6
Author: Matheus,
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/ajaxhispano.com/template/agent.layouts/content.php on line 61
2018-01-18 14:33:14