jon*_*boy 8 python group-by transform count pandas
我正在尝试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())
Run Code Online (Sandbox Code Playgroud)
这是我想要的输出:
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
Run Code Online (Sandbox Code Playgroud)
当我尝试执行类似操作但每月和每年执行一次时,请使用以下命令:
df1 = df.groupby([df['Date'].dt.year.rename('year'), df['Date'].dt.month.rename('month')]).agg({'count'})
print(df)
Run Code Online (Sandbox Code Playgroud)
但是输出是:
Date Val
count count
year month
2018 1 4 4
2 1 1
3 1 1
2019 1 2 2
Run Code Online (Sandbox Code Playgroud)
预期输出:
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
Run Code Online (Sandbox Code Playgroud)
使用GroupBy.transform与原来一样同数据帧大小列:
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
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
717 次 |
| 最近记录: |