将 NaN 文本列分隔到其他数据框中

Question

我有两列 X 和 Y，它们是文本列。 X 是我的特征，Y 是我的目标变量。我的 Y 变量由 NaN 值组成。我想将 Y 的 NaN 值分隔到一个单独的数据框中，以便在我的测试数据集中使用它们。

   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]