Переименуйте имена столбцов groupby и посчитайте результат с Pandas - PullRequest
Новая
       9

Переименуйте имена столбцов groupby и посчитайте результат с Pandas

1 голос
/ 10 марта 2020

Учитывая следующий фрейм данных: 

import numpy as np
df = pd.DataFrame({'price': np.random.random_integers(0, high=100, size=100)})
ranges = [0,10,20,30,40,50,60,70,80,90,100]
df.groupby(pd.cut(df.price, ranges)).count()

Out: 

          price
 price  
(0, 10]     9
(10, 20]    11
(20, 30]    11
(30, 40]    9
(40, 50]    16
(50, 60]    7
(60, 70]    10
(70, 80]    9
(80, 90]    14
(90, 100]   4

Как я могу reset_index результат и переименовать имена столбцов как bins и counts? Спасибо. 

      bins    counts
0   (0, 10]     9
1   (10, 20]    11
2   (20, 30]    11
3   (30, 40]    9
4   (40, 50]    16
5   (50, 60]    7
6   (60, 70]    10
7   (70, 80]    9
8   (80, 90]    14
9   (90, 100]   4

Ответы [ 2 ]

1 голос
/ 10 марта 2020

Одной из идей является использование rename для Серии от pd.cut, поэтому, если выбрать столбец price для групп обработки, будет вывод Series, поэтому добавьте Series.reset_index с параметром name для 2 columns DataFrame: 

df1 = (df.groupby(pd.cut(df.price, ranges).rename('bins'))['price'].count()
         .reset_index(name='counts'))
print (df1)
        bins  counts
0    (0, 10]      13
1   (10, 20]      13
2   (20, 30]       9
3   (30, 40]       9
4   (40, 50]       7
5   (50, 60]       9
6   (60, 70]       9
7   (70, 80]      12
8   (80, 90]       9
9  (90, 100]       9
0 голосов
/ 10 марта 2020

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

df.groupby(pd.cut(df.price, ranges)).count()\
.rename(columns={'price' : 'counts'})\
.reset_index()\
.rename(columns={'price': 'bins'})

Out: 

      bins    counts
0   (0, 10]     9
1   (10, 20]    11
2   (20, 30]    11
3   (30, 40]    9
4   (40, 50]    16
5   (50, 60]    7
6   (60, 70]    10
7   (70, 80]    9
8   (80, 90]    14
9   (90, 100]   4
...