如何基于熊猫的集合大小创建列

Question

我在pandas中有一个数据框，并且在groupby中使用“ set”作为聚合函数，并创建了以下数据框

A, B, C
1, 2, {nan, v}
1, 3, {nan, v,b}

在此示例中，A和B是我在groupby中使用的列。我想在此数据框中添加另一列，即列C中不是“ nan”的元素数。那就是我希望我的数据框如下所示

A, B, C, D
1, 2, {nan, v}, 1
1, 3, {nan, v,b}, 2

我该怎么做？

Answer 1

使用

df['D']=pd.DataFrame(df.C.tolist(),index=df.index).notnull().sum(axis=1)