Панды, переименовывая дубликат столбца внутри группового вызова

Question

Я использую панды, чтобы подсчитать элемент кадра данных.Мои данные выглядят примерно так:

A    B    C    
001  003  1
001  003  1
001  003  0 
...   

Я использую код:

exampleDataFrame = exampleDataFrame.groupby(['A','B','C'])['C'].size()

Где C может быть только 1 или 0. (Мне нужно посмотреть, сколько 1и 0 появляются в C для A и B).

Это дает мне такой результат:

A    B    C    C
001  003  1    2    (There were 2 '1's in C for columns A and B)
001  003  0    1
... 

Но теперь мне нужно переименовать 4-й столбец ('C') вболее уникальное имя, такое как «D».Каков наилучший способ сделать это?

Спасибо!

Answer 1

Попробуйте, rename ряд после группировки, затем reset_index для получения кадра данных:

df.groupby(['A','B','C'])['C'].size().rename('D').reset_index()

Вывод:

   A  B  C  D
0  1  3  0  1
1  1  3  1  2