pivot table pandas keepping nans code example

Example: pivot table but keep nan

(df.groupby(['Date', 'A']).B
   .apply(lambda x: np.nan if x.isna().all() else x.sum())
   .unstack('A')
)