python - 仅当所有元素都是 pandas 的 groupby 中的 NA 时，如何删除 NA

Question

我有一个看起来像这样的数据框

import pandas as pd
import numpy as np

fff = pd.DataFrame({'group': ['a','a','a','b','b','b','b','c','c'], 'value': [1,2, np.nan, 1,2,3,4, np.nan, np.nan]})

只有当 all value 是 group 内的 Nas 时，我才想按 group 删除 NA .我该怎么做？

预期输出:

fff = pd.DataFrame({'group': ['a','a','a','b','b','b','b'], 'value': [1,2, np.nan, 1,2,3,4]})

Answer 1

您可以检查 nan 的 value 并使用 groupby().any():

fff = fff[(~fff['value'].isna()).groupby(fff['group']).transform('any')]

输出:

  group  value
0     a    1.0
1     a    2.0
2     a    NaN
3     b    1.0
4     b    2.0
5     b    3.0
6     b    4.0