如何将groupby中的NaN元素放到pandas数据帧上?

使用自定义函数 dropna ,但对于空值,添加 NaN s:

print (my_df.groupby("D").agg(lambda x: np.nan if x.isnull().all() else x.dropna()))
        A    B     C
D                   
id1  gate  car  edge
id2  ball  NaN   NaN

具有自定义功能的类似解决方案:

def f(x):
    y = x.dropna()
    return np.nan if y.empty else y

print (my_df.groupby("D").agg(f))
        A    B     C
D                   
id1  gate  car  edge
id2  ball  NaN   NaN

我想你的方法要好得多,但最后添加了一个替换

my_df.fillna("").groupby("D").max().replace('',np.nan) 

您还可以执行以下操作:

def get_notnull(x):
    if x.notnull().any():
        return  x[x.notnull()]
    else:
        return np.nan

my_df.groupby('D').agg(get_notnull)

输出:

     A    B     C
D                   
id1  gate  car  edge
id2  ball  NaN   NaN