pandas聚合aggregate

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2018/5/24 15:03
# @Author  : zhang chao
# @File    : s.py
import pandas as pd
import numpy as np

df = pd.DataFrame(np.random.randn(10, 4),
      index = pd.date_range('1/1/2019', periods=10),
      columns = ['A', 'B', 'C', 'D'])

print (df)
print("=======================================")
r = df.rolling(window=3,min_periods=1)
print (r)
print("=======================================")
print("r.aggregate(np.sum)")
print (r.aggregate(np.sum))
print("=======================================")
print("r['A'].aggregate(np.sum)")
print (r['A'].aggregate(np.sum))
print("=======================================")
print("r[['A','B']].aggregate(np.sum)")
print (r[['A','B']].aggregate(np.sum))
print("=======================================")
print("r['A'].aggregate([np.sum,np.mean])")
print (r['A'].aggregate([np.sum,np.mean]))
print("=======================================")
print("r.aggregate({'A' : np.sum,'B' : np.mean})")
print (r.aggregate({'A' : np.sum,'B' : np.mean}))
print("=======================================")
print("r[['A','B']].aggregate([np.sum,np.mean]")
print (r[['A','B']].aggregate([np.sum,np.mean]))

D:\Download\python3\python3.exe D:/Download/pycharmworkspace/s.py
                   A         B         C         D
2019-01-01  0.744560  0.208652  0.542045 -0.995837
2019-01-02  0.029809 -1.419936 -0.461988  2.177032
2019-01-03  0.613583  1.515249  0.256546 -0.973564
2019-01-04  0.124320  1.152804  0.152107  1.629035
2019-01-05 -0.287906  1.003523 -0.793393  0.231969
2019-01-06 -0.045296 -0.921622  0.894335  0.773035
2019-01-07 -0.695347  0.512206  0.208833  0.953205
2019-01-08 -1.197178  0.142301 -0.854875 -1.044017
2019-01-09 -2.352468  0.047127 -0.351634 -0.373885
2019-01-10  0.678406  0.500947  0.304250 -0.606522
=======================================
Rolling [window=3,min_periods=1,center=False,axis=0]
=======================================
r.aggregate(np.sum)
                   A         B         C         D
2019-01-01  0.744560  0.208652  0.542045 -0.995837
2019-01-02  0.774369 -1.211283  0.080057  1.181195
2019-01-03  1.387952  0.303966  0.336603  0.207631
2019-01-04  0.767712  1.248117 -0.053335  2.832504
2019-01-05  0.449996  3.671576 -0.384740  0.887441
2019-01-06 -0.208882  1.234705  0.253049  2.634040
2019-01-07 -1.028549  0.594107  0.309775  1.958209
2019-01-08 -1.937820 -0.267115  0.248293  0.682223
2019-01-09 -4.244992  0.701633 -0.997676 -0.464698
2019-01-10 -2.871239  0.690374 -0.902259 -2.024425
=======================================
r['A'].aggregate(np.sum)
2019-01-01    0.744560
2019-01-02    0.774369
2019-01-03    1.387952
2019-01-04    0.767712
2019-01-05    0.449996
2019-01-06   -0.208882
2019-01-07   -1.028549
2019-01-08   -1.937820
2019-01-09   -4.244992
2019-01-10   -2.871239
Freq: D, Name: A, dtype: float64
=======================================
r[['A','B']].aggregate(np.sum)
                   A         B
2019-01-01  0.744560  0.208652
2019-01-02  0.774369 -1.211283
2019-01-03  1.387952  0.303966
2019-01-04  0.767712  1.248117
2019-01-05  0.449996  3.671576
2019-01-06 -0.208882  1.234705
2019-01-07 -1.028549  0.594107
2019-01-08 -1.937820 -0.267115
2019-01-09 -4.244992  0.701633
2019-01-10 -2.871239  0.690374
=======================================
r['A'].aggregate([np.sum,np.mean])
                 sum      mean
2019-01-01  0.744560  0.744560
2019-01-02  0.774369  0.387185
2019-01-03  1.387952  0.462651
2019-01-04  0.767712  0.255904
2019-01-05  0.449996  0.149999
2019-01-06 -0.208882 -0.069627
2019-01-07 -1.028549 -0.342850
2019-01-08 -1.937820 -0.645940
2019-01-09 -4.244992 -1.414997
2019-01-10 -2.871239 -0.957080
=======================================
r.aggregate({'A' : np.sum,'B' : np.mean})
                   B         A
2019-01-01  0.208652  0.744560
2019-01-02 -0.605642  0.774369
2019-01-03  0.101322  1.387952
2019-01-04  0.416039  0.767712
2019-01-05  1.223859  0.449996
2019-01-06  0.411568 -0.208882
2019-01-07  0.198036 -1.028549
2019-01-08 -0.089038 -1.937820
2019-01-09  0.233878 -4.244992
2019-01-10  0.230125 -2.871239
=======================================
r[['A','B']].aggregate([np.sum,np.mean]
                   A                   B          
                 sum      mean       sum      mean
2019-01-01  0.744560  0.744560  0.208652  0.208652
2019-01-02  0.774369  0.387185 -1.211283 -0.605642
2019-01-03  1.387952  0.462651  0.303966  0.101322
2019-01-04  0.767712  0.255904  1.248117  0.416039
2019-01-05  0.449996  0.149999  3.671576  1.223859
2019-01-06 -0.208882 -0.069627  1.234705  0.411568
2019-01-07 -1.028549 -0.342850  0.594107  0.198036
2019-01-08 -1.937820 -0.645940 -0.267115 -0.089038
2019-01-09 -4.244992 -1.414997  0.701633  0.233878
2019-01-10 -2.871239 -0.957080  0.690374  0.230125

Process finished with exit code 0

猜你喜欢

转载自www.cnblogs.com/ggzhangxiaochao/p/9094251.html