熊猫:不要把南计算在一个集合函数中。

Question

我有以下代码：

data_agg_df = data_df.groupby("team", as_index=False).player.agg({"player_set": lambda x: set(list(x)), "player_count": "nunique"})

那么我的结果看起来是：

team           player_set         player_count
-------------------------------------------------
 A           {John, Mary}          2
 B           {nan}                 0
 C           {Dave,nan}            1

我想知道怎样才能不让娜娜在player_set上露面呢？也就是说，我希望得到的数据框架如下所示：

team           player_set         player_count
-------------------------------------------------
 A           {John, Mary}          2
 B           {}                    0
 C           {Dave}                1

谢谢!

Answer 1

替换

set(list(x))

使用

set(list(i for i in x if pd.notnull(i)))

取出nans