|
参考
- import pandas as pd
- def get_maxdata(x):
- stock_data=pd.DataFrame()
- df=pd.read_csv(x,dtype={'trade_date':'str'})
- df['trade_date']=pd.to_datetime(df['trade_date'])
- df['year']=df['trade_date'].dt.year
- max_row=df[df['year']==2020]['close'].idxmax()
- end_date_max=df.iloc[max_row]['trade_date']
- start_date_max=end_date_max-pd.Timedelta(days=100)
- maxdf=df[(df['trade_date']>=start_date_max) & (df['trade_date']<=end_date_max)]
- stock_data=stock_data.append(maxdf,ignore_index=True)
- stock_data.to_excel('reuslt.xlsx',index=False)
- get_maxdata('000001.SZ.csv')
复制代码 |
|