Python学习日记之从Tushare上获取500成分股

系列文章目录

第一章 Python 从 Tushare 导入中证500成分股

---

1.下载Tushare

import tushare as ts
import pandas as pd
import os

2.指定路径相关设置

os.chdir('D:\ChengFenGu_ts_code/')  #保存的绝对路径
pro = ts.pro_api('pro = ts.pro_api('your token')')
pd.set_option('expand_frame_repr', False)#True就是可以换行显示。设置成False的时候不允许换行
pd.set_option('display.max_columns', None)# 显示所有列
#pd.set_option('display.max_rows', None)# 显示所有行
pd.set_option('colheader_justify', 'centre')# 显示居中

3.定义函数

pro = ts.pro_api()
def ZZ500ChengFenHS(index_code, trade_date):
    ZZ500ChengFen = pro.index_weight(index_code=index_code, trade_date=trade_date)
    file_path = 'D:\TiaoZheng1ShuJu\ChengFenGu_ts_code'
    if not os.path.exists(file_path):
        os.makedirs(file_path)
    # 完整的文件+路径名
    #date_fmt = arrow.now('local').format("YYYY-MM-DD_HH-mm_ss")
    file_name = trade_date+'_ZhongZheng500ChengFenDaiMa.csv'
    # 写入内容
    ZZ500ChengFen.to_csv(
        os.path.join(file_path, file_name),index=False)

4.调用函数

index_code='000905.SH'
trade_date='20220331'
ZZ500ChengFenHS(index_code, trade_date)

4.小结

每次使用只需输入index_code和 trade_date两个参数即可的到500成分股代码。