Python pandas Фильтрация Nan из выборки данных столбца строк


190

Без использования, groupbyкак бы я отфильтровать данные без NaN?

Допустим, у меня есть матрица, в которой клиенты будут заполнять 'N / A', 'N / A' или любой из его вариантов, а другие оставляют это поле пустым:

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) ]

вывод:

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

Как бы я отфильтровал значения NaN, чтобы получить результаты для работы следующим образом:

  movie    name  rating
0   thg    John       3
3   mol  Graham     NaN

Я думаю, мне нужно что-то вроде, ~np.isnanно тильда не работает со строками.

Ответы:


257

Просто бросьте их:

nms.dropna(thresh=2)

это отбросит все строки, где есть как минимум два не- NaN.

Тогда вы могли бы затем бросить, где имя 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]

РЕДАКТИРОВАТЬ

На самом деле, глядя на то, что вы изначально хотели, вы можете сделать это без dropnaзвонка:

nms[nms.name.notnull()]

ОБНОВИТЬ

Глядя на этот вопрос спустя 3 года, возникает ошибка, во-первых, thresharg ищет хотя бы nненулевые NaNзначения, поэтому на самом деле результат должен быть:

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

Вполне возможно, что я либо ошибся 3 года назад, либо в версии панд, с которой я работал, была ошибка, оба сценария вполне возможны.


194

Самое простое из всех решений:

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

Таким образом, он отфильтровывает только те строки, которые не имеют значений NaN в столбце «имя».

Для нескольких столбцов:

filtered_df = df[df[['name', 'country', 'region']].notnull().all(1)]

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])]

Используя наш сайт, вы подтверждаете, что прочитали и поняли нашу Политику в отношении файлов cookie и Политику конфиденциальности.
Licensed under cc by-sa 3.0 with attribution required.