разделить текстовые столбцы NaN на другой фрейм данных

Question

У меня есть два столбца X и Y, которые являются текстовыми столбцами.X - это моя особенность, а Y - моя целевая переменная.Моя переменная Y состоит из NaN значений.Я хочу разделить NaN значения Y в отдельный фрейм данных, чтобы использовать их в своем тестовом наборе данных.

   Y
1. aaa
2. bbb
3. ccc
4. NaN
5. NaN
6. ddd

Может кто-нибудь сказать мне, как разделить строки NaN?

Answer 1

Я считаю, что нужно boolean indexing с isna:

mask = df['Y'].isna()
#oldier pandas versions
#mask = df['Y'].isnull()
#DataFrame with NaNs
df1 = df[mask]
#DataFrame with no NaNs
df2 = df[~mask]