Python pandas filtrant nan à partir d'une sélection de données d'une colonne de chaînes

190

Sans utiliser, groupbycomment puis-je filtrer les données sans NaN?

Supposons que j'ai une matrice dans laquelle les clients rempliront «N / A», «n / a» ou l'une de ses variantes et que d'autres la laisseront vide:

import pandas as pd
import numpy as np


df = pd.DataFrame({'movie': ['thg', 'thg', 'mol', 'mol', 'lob', 'lob'],
                  'rating': [3., 4., 5., np.nan, np.nan, np.nan],
                  'name': ['John', np.nan, 'N/A', 'Graham', np.nan, np.nan]})

nbs = df['name'].str.extract('^(N/A|NA|na|n/a)')
nms=df[(df['name'] != nbs) ]

production:

>>> nms
  movie    name  rating
0   thg    John       3
1   thg     NaN       4
3   mol  Graham     NaN
4   lob     NaN     NaN
5   lob     NaN     NaN

Comment filtrer les valeurs NaN afin que je puisse obtenir des résultats avec lesquels travailler comme ceci:

  movie    name  rating
0   thg    John       3
3   mol  Graham     NaN

Je suppose que j'ai besoin de quelque chose comme ~np.isnanmais le tilda ne fonctionne pas avec des cordes.

ccsv
la source

Réponses:

257

Laissez-les simplement tomber:

nms.dropna(thresh=2)

cela supprimera toutes les lignes où il y a au moins deux non- NaN.

Ensuite, vous pouvez déposer où se trouve le nom NaN:

In [87]:

nms
Out[87]:
  movie    name  rating
0   thg    John       3
1   thg     NaN       4
3   mol  Graham     NaN
4   lob     NaN     NaN
5   lob     NaN     NaN

[5 rows x 3 columns]
In [89]:

nms = nms.dropna(thresh=2)
In [90]:

nms[nms.name.notnull()]
Out[90]:
  movie    name  rating
0   thg    John       3
3   mol  Graham     NaN

[2 rows x 3 columns]

ÉDITER

En fait, en regardant ce que vous voulez à l'origine, vous pouvez le faire sans l' dropnaappel:

nms[nms.name.notnull()]

METTRE À JOUR

En regardant cette question 3 ans plus tard, il y a une erreur, premièrement thresharg recherche au moins des nnon- NaNvaleurs donc en fait la sortie devrait être:

In [4]:
nms.dropna(thresh=2)

Out[4]:
  movie    name  rating
0   thg    John     3.0
1   thg     NaN     4.0
3   mol  Graham     NaN

Il est possible que je me sois trompé il y a 3 ans ou que la version de pandas que j'utilisais avait un bug, les deux scénarios sont tout à fait possibles.

EdChum
la source
194

La plus simple de toutes les solutions:

filtered_df = df[df['name'].notnull()]

Ainsi, il filtre uniquement les lignes qui n'ont pas de valeurs NaN dans la colonne «nom».

Pour plusieurs colonnes:

filtered_df = df[df[['name', 'country', 'region']].notnull().all(1)]
Gil Baggio
la source
9
df = pd.DataFrame({'movie': ['thg', 'thg', 'mol', 'mol', 'lob', 'lob'],'rating': [3., 4., 5., np.nan, np.nan, np.nan],'name': ['John','James', np.nan, np.nan, np.nan,np.nan]})

for col in df.columns:
    df = df[~pd.isnull(df[col])]
Bashar Mohammad
la source
5
df.dropna(subset=['columnName1', 'columnName2'])
JacoSolari
la source