Как создать столбец на основе размера набора в пандах

Question

У меня есть датафрейм в пандах, и я использую '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)