这里我们选取的是中国巨石这一只股票,也可以通过更改参数去获取别的股票的行情。
没有对应模块的下载模块再运行:
pip install tushare
pip install pandas
pip install mplfinance
废话不多说上代码:
# -*- codeing = utf-8 -*-
# @Time : 2021/9/1 21:27
# @Author : 彭翔
# @File : tushare_test.py
# @Software: PyCharm
import pandas as pd
import tushare as ts
import mplfinance as mpf
# 设置token
# 这里的token需要自己去tushare官网获取哦
pro = ts.pro_api('63e7d81e797242364243e3116641934823899f1e77459848ba7d6be0')
# 查找需要获取数据的股票
data_name = pro.stock_basic(exchange='', list_status='L', name='中国巨石', fields='ts_code,symbol,name,area,industry,'
'list_date')
# 获取我们需要的中国巨石股票的五年内的数据
df = pro.daily(ts_code='600176.SH', start_date='20170903', end_date='20210903',
fields='trade_date,open,high,low,close,vol')
df['trade_date'] = pd.to_datetime(df['trade_date'])
# 放置的路径
df.to_csv('600176.SH.csv')
df.to_excel('600176.SH.xls')
# 设置dataframe的列名
col_names = ["Date", "Open", "High", "Low", "Close", "Volume"]
# 获取dataframe的数据
data = pd.read_csv(r"600176.SH.csv", header=0, names=col_names)
data = data.iloc[::-1]
# 修改第一行的日期格式
data.timestamp = pd.to_datetime(data.Date, format='%Y-%m-%d')
# 将第一行设置为时间索引格式
data.index = data.timestamp
data.drop('Date', axis=1, inplace=True)
data.head()
# 使用mpf里的方法去绘制日线图
mpf.plot(data, type='candle', mav=(2, 5, 10), volume=True)
效果图: