"""
2020.12.01
封裝成函式
"""
import pandas as pd
import tushare as ts
def data_get(date):
token = 'cf5e9df84cde5b1246db7dc1ba08d36d2e4a7760f6f078fc4c4b823c'
pro = ts.pro_api(token)
dfx = pro.share_float(float_date=date)
return dfx
def func():
date = 20201201
df = data_get(date)
df1 = pd.DataFrame(df.ts_code.str[0:6])
new_stock = df1['ts_code'].drop_duplicates()
new_stock.to_excel('003.xlsx', encoding='utf-8')
df2 = pd.DataFrame(pd.read_excel('003.xlsx', sheet_name='Sheet1'))
df3 = pd.DataFrame(pd.read_excel('C:\\Users\\SINBAD\\Desktop\\all_Astock.xlsx', sheet_name='Sheet1'))
df_merge = pd.merge(df2, df3, how='inner', on='ts_code')
df_merge.head()
df_merge.to_excel('F:\\限售解禁\\20201201.xlsx', sheet_name='Sheet1', index=False)
try:
for i in range(1, 29):
df4 = pd.DataFrame(pd.read_excel('F:\\限售解禁\\20201201.xlsx', sheet_name='Sheet1'))
date += 1
df_ = data_get(date)
df1_ = pd.DataFrame(df_.ts_code.str[0:6])
new_stock_ = df1_['ts_code'].drop_duplicates()
new_stock_.to_excel('003.xlsx', encoding='utf-8')
df2_ = pd.DataFrame(pd.read_excel('003.xlsx', sheet_name='Sheet1'))
df_merge_ = pd.merge(df2_, df3, how='inner', on='ts_code')
df_merge_.head()
df_merge_merge_ = pd.merge(df4, df_merge_, how='outer')
df_merge_merge_.to_excel('F:\\限售解禁\\20201201.xlsx', sheet_name='Sheet1', index=False)
i += 1
else:
print('finish')
except:
print('byebye')
if __name__ == '__main__':
func()
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/228560.html
上一篇：pycharm安裝第三方庫報錯