Я пытаюсь groupby подсчитать даты в месяц и год в конкретном выводе. Я могу делать это в день, но не могу получить такой же результат в месяц/год.
d = ({
'Date' : ['1/1/18','1/1/18','2/1/18','3/1/18','1/2/18','1/3/18','2/1/19','3/1/19'],
'Val' : ['A','B','C','D','A','B','C','D'],
})
df = pd.DataFrame(data = d)
df['Date'] = pd.to_datetime(df['Date'], format= '%d/%m/%y')
df['Count_d'] = df.Date.map(df.groupby('Date').size())
Это результат, который я хочу:
Date Val Count_d
0 2018-01-01 A 2
1 2018-01-01 B 2
2 2018-01-02 C 1
3 2018-01-03 D 1
4 2018-02-01 A 1
5 2018-03-01 B 1
6 2019-01-02 C 1
7 2019-01-03 D 1
Когда я пытаюсь сделать то же самое, но в месяц и год, я использую следующее:
df1 = df.groupby([df['Date'].dt.year.rename('year'), df['Date'].dt.month.rename('month')]).agg({'count'})
print(df)
Но вывод:
Date Val
count count
year month
2018 1 4 4
2 1 1
3 1 1
2019 1 2 2
Предполагаемый результат:
Date Val Count_d Count_m Count_y
0 2018-01-01 A 2 4 6
1 2018-01-01 B 2 4 6
2 2018-01-02 C 1 4 6
3 2018-01-03 D 1 4 6
4 2018-02-01 A 1 1 6
5 2018-03-01 B 1 1 6
6 2019-01-02 C 1 2 2
7 2019-01-03 D 1 2 2






Используйте GroupBy.transform для столбцов того же размера, что и исходный DataFrame:
df['Date'] = pd.to_datetime(df['Date'], format= '%d/%m/%y')
y = df['Date'].dt.year
m = df['Date'].dt.month
df['Count_d'] = df.groupby('Date')['Date'].transform('size')
df['Count_m'] = df.groupby([y, m])['Date'].transform('size')
df['Count_y'] = df.groupby(y)['Date'].transform('size')
print(df)
Date Val Count_d Count_m Count_y
0 2018-01-01 A 2 4 6
1 2018-01-01 B 2 4 6
2 2018-01-02 C 1 4 6
3 2018-01-03 D 1 4 6
4 2018-02-01 A 1 1 6
5 2018-03-01 B 1 1 6
6 2019-01-02 C 1 2 2
7 2019-01-03 D 1 2 2
@anky_91 - потому что столбцы того же размера, что и исходный df.
где ты это увидел @anky_91
@Erfan получил предупреждение на будущее. Я неправильно реализовал, я думаю, Джез ясно дал понять
Вы можете сделать это с pd.Grouper
df['Count_d'] = df.groupby([pd.Grouper(key='Date', freq='D')])['Date'].transform('size').astype(int)
df['Count_m'] = df.groupby([pd.Grouper(key='Date', freq='M')])['Date'].transform('size').astype(int)
df['Count_y'] = df.groupby([pd.Grouper(key='Date', freq='Y')])['Date'].transform('size').astype(int)
Что даст
Date Val Count_d Count_m Count_y
0 2018-01-01 A 2 4 6
1 2018-01-01 B 2 4 6
2 2018-01-02 C 1 4 6
3 2018-01-03 D 1 4 6
4 2018-02-01 A 1 1 6
5 2018-03-01 B 1 1 6
6 2019-01-02 C 1 2 2
7 2019-01-03 D 1 2 2
С этим вы можете группировать по разным частотам, см. документация по DateOffsets
только что обнаружил, что они удаляют gg с помощью dict. есть идеи, почему?