после использования reset_index от groupby & count на DF не может переименовать столбец count - PullRequest
0 голосов
/ 02 июля 2018

Это мой код ниже, так как я выполнил групповую работу и рассчитывал на большой фрейм данных

group3=charity.groupby(['Split', 'B4']).size()
group3=group3.reset_index()

out
       split     B4     0
  0     CRk      No    193
  1     CRuk     Yes    7
  2     LLR      No    184
  3     LLR      Yes    15
  4     MUK      No     188
  5     MUK      Yes    12
  6     MCUK     No    186
  7     MCUK     Yes    14

Ниже код не работает

  group3=group3.rename(columns={"0": "count1"})
  group3 #does not work 

хочу переименовать новый столбец 0, созданный функцией count, чтобы я мог добавить другие столбцы во фрейм данных для теста хи-квадрат

также интересно, как я могу сделать функцию подсчета для нескольких столбцов в дополнение к столбцу B4 выше

1 Ответ

0 голосов
/ 02 июля 2018

Использовать имя параметра в reset_index:

group3=charity.groupby(['Split', 'B4']).size().reset_index(name='count1')

Или переименовать скалярное число 0:

group3 = group3.rename(columns={0: "count1"})

Sample

charity = pd.DataFrame({'B4':list('abbbbb'),
                       'Split':list('aaabbb')})

print (charity)
  B4 Split
0  a     a
1  b     a
2  b     a
3  b     b
4  b     b
5  b     b

group3=charity.groupby(['Split', 'B4']).size().reset_index(name='count1')
print (group3)
  Split B4  count1
0     a  a       1
1     a  b       2
2     b  b       3
...