python笔记：3.2.2.5pandas数据操作

# -*- 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