У меня есть фреймворк в пандах, и я использую 'set' в качестве агрегатной функции в groupby, и я создаю следующий фреймворк

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

Как я могу это сделать?

0
HHH 15 Ноя 2018 в 18:27

1 ответ

Лучший ответ

С помощью

df['D']=pd.DataFrame(df.C.tolist(),index=df.index).notnull().sum(axis=1)
1
BENY 15 Ноя 2018 в 15:39