用Python编写程序,采用tushare库,设置 token,使用 ts.pro_api() 接口,获取所有申万二级行业指数代码最近三个月的行情数据,储存到文件
时间: 2024-03-22 18:39:25 浏览: 74
好的,我会为您完成这个任务。
首先,您需要安装 tushare 库。可以使用以下命令在终端中安装:
```
pip install tushare
```
然后,您需要在 tushare 官网上注册账号并获取 token。在代码中设置 token,代码如下:
```python
import tushare as ts
token = 'YOUR_TOKEN_HERE'
pro = ts.pro_api(token)
```
接下来,使用 `pro.index_classify()` 方法获取申万二级行业指数代码:
```python
sw_index = pro.index_classify(level='L2', src='SW')
index_codes = sw_index['index_code'].tolist()
```
然后,使用 `pro.index_daily()` 方法获取最近三个月的行情数据,并将数据储存到文件中:
```python
import os
import pandas as pd
output_dir = 'output'
if not os.path.exists(output_dir):
os.makedirs(output_dir)
for code in index_codes:
df = pro.index_daily(ts_code=code, start_date='20211201', end_date='20220301')
filename = os.path.join(output_dir, f'{code}.csv')
df.to_csv(filename, index=False)
```
上述代码中,`start_date` 和 `end_date` 参数是指最近三个月的时间范围,这里设置为 2021 年 12 月 1 日至 2022 年 3 月 1 日。
最后,运行代码即可完成任务。数据将储存在 output 文件夹中,每个文件以申万二级行业指数代码命名,为 csv 格式。
阅读全文