python笔记：3.2.2.5pandas数据操作_分组_对20190523股市数据进行简单描述统计分析

# -*- coding: utf-8 -*-
"""
Created on Thu May 23 16:17:49 2019

@author: User
"""

import tushare as ts

ts.set_token('***输入自己的token')
pro = ts.pro_api()

#df = pro.daily(ts_code='000001.SZ', start_date='20180701', end_date='20180718')
df = pro.daily(trade_date='20190523')

print(df.columns)
print(df.head())

print('\n删掉几列：')
df.drop(['high','low','pre_close','change','pct_chg','vol','amount'],axis=1,inplace=True)
print(df.columns)

print('\n 增加一列，收盘价>开盘价则变量标记为GOOD，否则为BAD')
df['Market']=list(map(lambda x: 'Good' if x>0 else 
                     ('Bad' if x<0 else 'OK' ),
                     df['close']-df['open']))
print(df.head())

print('\n 分组：')
dfgrouped=df.groupby(df['Market'])
print(dfgrouped.head())

print('\n 对开盘价进行统计分析：')
print(dfgrouped['open'].describe())

运行：

Index(['ts_code', 'trade_date', 'open', 'high', 'low', 'close', 'pre_close',
       'change', 'pct_chg', 'vol', 'amount'],
      dtype='object')
     ts_code trade_date   open   high  ...  change  pct_chg        vol      amount
0  000736.SZ   20190523   9.81   9.81  ...   -0.33  -3.3639   34163.50   32690.838
1  000737.SZ   20190523   3.11   3.18  ...    0.03   0.9524   26995.10    8471.132
2  000738.SZ   20190523  14.01  14.28  ...   -0.04  -0.2855   64795.87   91251.070
3  000739.SZ   20190523   9.91   9.91  ...   -0.34  -3.4240   78157.00   75467.558
4  000750.SZ   20190523   4.86   5.03  ...    0.03   0.6122  703426.05  348299.607

[5 rows x 11 columns]

删掉几列：
Index(['ts_code', 'trade_date', 'open', 'close'], dtype='object')

 增加一列，收盘价>开盘价则变量标记为GOOD，否则为BAD
     ts_code trade_date   open  close Market
0  000736.SZ   20190523   9.81   9.48    Bad
1  000737.SZ   20190523   3.11   3.18   Good
2  000738.SZ   20190523  14.01  13.97    Bad
3  000739.SZ   20190523   9.91   9.59    Bad
4  000750.SZ   20190523   4.86   4.93   Good

 分组：
       ts_code trade_date   open  close Market
0    000736.SZ   20190523   9.81   9.48    Bad
1    000737.SZ   20190523   3.11   3.18   Good
2    000738.SZ   20190523  14.01  13.97    Bad
3    000739.SZ   20190523   9.91   9.59    Bad
4    000750.SZ   20190523   4.86   4.93   Good
5    000751.SZ   20190523   3.38   3.31    Bad
6    000752.SZ   20190523   5.02   4.93    Bad
11   000758.SZ   20190523   5.15   5.18   Good
13   000760.SZ   20190523   1.67   1.67     OK
14   000761.SZ   20190523   4.39   4.43   Good
23   000780.SZ   20190523   3.61   3.68   Good
25   000783.SZ   20190523   7.10   7.10     OK
91   600614.SH   20190523   2.70   2.70     OK
116  600643.SH   20190523   9.65   9.65     OK
125  600653.SH   20190523   2.25   2.25     OK

 对开盘价进行统计分析：
         count       mean        std   min     25%     50%      75%     max
Market                                                                     
Bad     3046.0  13.795318  21.503680  1.11  5.6700   9.145  15.9975  881.00
Good     477.0  13.134172  11.390314  1.82  6.2200  10.140  16.2600   96.20
OK        70.0   9.502571  15.883855  1.23  3.0525   5.205  10.2525  121.67