python - 如果包含单个 NaN 并组合列，则将整个组设置为 NaN

Question

我有一个df

a  b  c    d
0  1  nan  1
0  2  2    nan     
0  2  3    4
1  3  1    nan
1  1  nan  3
1  1  2    3
1  1  2    4

我需要按 a 和 b 进行分组，然后如果 c 或 d 在组内包含 1 个或多个 nan，我希望特定列中的整个组都是 nan:

a  b  c    d
0  1  nan  1
0  2  2    nan     
0  2  3    nan
1  3  1    nan
1  1  nan  3
1  1  nan  3
1  1  nan  4

然后将c和d组合起来，就不再有nan了

a  b  c    d    e
0  1  nan  1    1
0  2  2    nan  2   
0  2  3    nan  3
1  3  1    nan  1
1  1  nan  3    3
1  1  nan  3    3
1  1  nan  4    4

Answer 1

您需要检查每个组是否为 nan，然后设置适当的值(nan 或现有值)，然后使用 combine_first() 组合列。

from io import StringIO
import pandas as pd
import numpy as np
df = pd.read_csv(StringIO("""
a b c d
0 1 nan 1
0 2 2 nan
0 2 3 4
1 3 1 nan
1 1 nan 3
1 1 2 3
1 1 2 4
"""), sep=' ')

for col in ['c', 'd']:
    df[col] = df.groupby(['a','b'])[col].transform(lambda x: np.nan if any(x.isna()) else x)

df['e'] = df['c'].combine_first(df['d'])
df
    a   b   c   d   e
0   0   1   NaN 1.0 1.0
1   0   2   2.0 NaN 2.0
2   0   2   3.0 NaN 3.0
3   1   3   1.0 NaN 1.0
4   1   1   NaN 3.0 3.0
5   1   1   NaN 3.0 3.0
6   1   1   NaN 4.0 4.0