python - 按行分组、NaN 值和数字之间的联接和选择

Question

我想转换这个数据框:

A   B   C   D
xx  nan 2   3
xx  1   nan 2
yy  4   nan nan
yy  nan 5   nan
yy  nan 4   6
zz  nan nan nan
zz  nan 8   nan
zz  nan 9   nan
zz  7   10  nan

进入最后一个(按“A”列分组)

A   B   C   D
xx  1   2   3
yy  4   5   6
zz  7   8   nan

我尝试使用groupby、fillna和apply，但到目前为止没有结果。

我希望最终的数据框尽可能显示非 nan 值以及每个唯一“A”列值的第一个值。

我怎样才能做到这一点？

Answer 1

使用groupby + first :

df = df.groupby('A', as_index=False).first()

替代方案是 agg具有自定义功能，如果将来会解决 this issue :

def f(x):
    y = x.dropna()
    return np.nan if y.empty else y.iat[0]
df = df.groupby('A', as_index=False).agg(f)

print (df)
    A    B    C    D
0  xx  1.0  2.0  3.0
1  yy  4.0  5.0  6.0
2  zz  7.0  8.0  NaN