一行代码获取股票、基金数据，并绘制K线图

小知识，大挑战！本文正在参与“ 程序员必备小知识 ”创作活动

在本文同时参与 「掘力星计划」 ，赢取创作大礼包，挑战创作激励金

我是老表，Python终身学习者，数据分析爱好者，宠猫狂人～

今天这篇文章和大家分享一下如何利用Python获取股票、基金数据，并进行可视化。

一、基础准备

环境：python 3.7+
需要安装第三方模块：mplfinance、akshare
运行编辑器：jupyter notebook

二、动手动脑

2.1 环境准备

首先需要你的电脑安装好了Python环境，并且安装好了Python开发工具。

如果你还没有安装，可以参考以下文章：

如果仅用Python来处理数据、爬虫、数据分析或者自动化脚本、机器学习等，建议使用Python基础环境+jupyter即可，安装使用参考Windows/Mac 安装、使用Python环境+jupyter notebook

2.2 模块安装

本次实验是在jupyter notebook上进行，默认你已经安装好了Python基本环境和jupyter，如果没有安装可以看上面的环境安装部分。

首先打开终端，输入：

jupyter notebook
复制代码

即可进入jupyter，会自动连接服务器，并跳转到网页，新建一个python文件。给文件重命名，容易区分：

在代码块中输入以下指令并运行，即可安装mplfinance（金融数据可视化，基于matplotlibe和pandas）和akshare（金融数据获取模块）：

!pip3 install mplfinance --user
!pip3 install akshare --user
复制代码

2.3 模块基本介绍与使用

2.3.1 mplfinance
基本介绍：用于金融数据可视化和可视化分析的 matplotlib 实用程序（接口），基于matplotlibe开发，并且与pandas的DataFrame数据有很好的结合。

项目地址：github.com/matplotlib/…

基本使用方法：

import mplfinance as mpf

mpf.plot(data)
复制代码

其中 data数据是一个DataFrame数据类型，必须包含：Open, High, Low and Close这四列，并且有时间日期索引（类型为：DatetimeIndex）。

接下来，我们虚拟一个数据来看看效果吧：

import mplfinance as mpf
import pandas as pd

# 创建日期索引
index_date = pd.DatetimeIndex(['20210906', '20210907', '20210908', '20210909'])
# 随便创建数据
data = pd.DataFrame({
    'Open': [1, 2, 3, 4], 
    'High': [2, 3, 4, 5], 
    'Low' : [0.5, 1, 2, 2],
    'Close': [2, 3, 4, 3]
}, index=index_date)
# 调用plot函数 进行绘图
mpf.plot(data)
复制代码

我们看到，现在根据模拟数据我们简单绘制了一个简单的OHLC线条。

type参数使用，让绘图更加精彩：

mpf.plot(data,type='candle')
复制代码

mpf.plot(data,type='line')
复制代码

type还可以为：renko、pnf。

mav参数可以添加移动平均线，更好的看趋势。

mpf.plot(data, type='candle', mav=2)
复制代码

还可以添加多条移动平均线：

mpf.plot(data, type='candle', mav=(2, 3))
复制代码

更多使用方法，后面分章节给大家介绍，也欢迎大家去github地址自行学习，和我以及简说Python学习群友进行交流。

2.3.2 akshare
基本介绍：AKShare 是一个优雅简单的 Python 金融数据接口库，你可以利用这个库轻松获取到金融财经数据。

项目地址：github.com/jindaxiang/… 基本使用方法：

获取指数数据

import akshare as ak
# 获取上证指数每日的变化数据 OHLC
sz_index = ak.stock_zh_index_daily(symbol="sh000001")
复制代码

这样就可以获取到上证指数所有历史数据啦～

获取A股数据

# 获取茅台股票每日的变化数据 采用前复权方式计算（关于复权计算更多知识以后分享，大家可以网络查找相关资料）
stock_zh_mt_hist_df = ak.stock_zh_a_hist(symbol="600519", period="daily", start_date="20170301", end_date='20210913', adjust="qfq")
复制代码

获取港股数据

# 获取港股腾讯股票历史数据
stock_hk_tx_hist_df = ak.stock_hk_hist(symbol="00700", start_date="20170301", end_date="20210913", adjust="qfq")
复制代码

获取美股数据

# 获取美股苹果公司历史股票数据
stock_us_apple_hist_df = ak.stock_us_hist(symbol='105.AAPL', start_date="20100101", end_date="20210913", adjust="qfq")
复制代码

获取基金数据

# 获取易方达蓝筹从发行到现在的净值数据
yfd_fund = ak.fund_em_open_fund_info(fund="005827", indicator="单位净值走势")
复制代码

更多使用方法，后面分章节给大家介绍，也欢迎大家去github地址自行学习，和我以及简说Python学习群友进行交流。

来，操作

按上面方法，我们获取上证指数数据后，选取从2020-01-01到现在的数据进行可视化，然后在进行细分可视化，并选择不同的移动平均线。

2020-01-01 - '2021-09-13， mav=(200, 300, 350)

import akshare as ak
import mplfinance as mpf
import pandas as pd

# 获取上证指数每日的变化数据 OHLC
sz_index = ak.stock_zh_index_daily(symbol="sh000001")
data1 = sz_index.loc['2020-01-01 00:00:00+00:00':'2021-09-13 00:00:00+00:00']
mpf.plot(data1, type='candle', mav=(200, 300, 350), volume=True)
复制代码

2021-01-01 - '2021-09-13， mav=(30, 60, 120)

data2 = sz_index.loc['2021-01-01 00:00:00+00:00':'2021-09-13 00:00:00+00:00']
mpf.plot(data2, type='candle', mav=(30, 60, 120), volume=True)
复制代码

2021-04-01 - '2021-09-13， mav=(5, 10, 20)

data3 = sz_index.loc['2021-04-01 00:00:00+00:00':'2021-09-13 00:00:00+00:00']
mpf.plot(data3, type='candle', mav=(5, 10, 20), volume=True)
复制代码

今天就到这里啦，如果大家觉得文章不错，记得点个赞哈～金融分析&可视化系列会继续，本文全部源码还在整理中，大家可以先加老表微信，备注：金融分析，加入金融分析&可视化交流群，后面代码整理好了会第一时间群内公布获取方法。

点赞+留言+转发，就是对我最大的支持啦～