用Python打造你的股市分析工具箱:从数据获取到K线图绘制

余汉波看波涛 2024-02-25 02:27:15

K 线图是金融市场中常用的技术分析工具,可以直观地展示股票价格的走势和波动情况。本文将介绍如何使用 Python 和 akshare、mplfinance 库来获取东方财富网的股票日线数据并绘制 K 线图。

代码import akshare as akimport mplfinance as mpfimport pandas as pdimport matplotlib.pyplot as plt# 设置 Matplotlib 支持中文显示plt.rcParams['font.sans-serif'] = ['Microsoft YaHei'] # 设置中文字体,确保中文能够正常显示plt.rcParams['axes.unicode_minus'] = False # 解决负号 '-' 显示为方块的问题def get_stock_data_eastmoney(symbol, start_date, end_date): # 使用 akshare 的 stock_zh_a_hist 函数获取东方财富网的股票日线数据 df = ak.stock_zh_a_hist(symbol=symbol, period="daily", start_date=start_date, end_date=end_date, adjust="qfq") # 检查获取的 DataFrame 是否为空 if df.empty: print("未获取到数据,请检查股票代码及日期范围是否正确。") return # 将日期列设置为索引,并转换为 datetime 类型 df['日期'] = pd.to_datetime(df['日期']) df.set_index('日期', inplace=True) # 调整 DataFrame 列名以符合 mplfinance 的要求 df.rename(columns={ '开盘': 'Open', '收盘': 'Close', '最高': 'High', '最低': 'Low', '成交量': 'Volume' }, inplace=True) # 转换列名为小写,以符合 mplfinance 的要求 df.columns = df.columns.str.lower() # 定义 mplfinance 的自定义风格 mc = mpf.make_marketcolors(up='r', down='g', volume='inherit') s = mpf.make_mpf_style(base_mpf_style='charles', marketcolors=mc, rc={'font.sans-serif': ['Microsoft YaHei']}) # 使用 mplfinance 绘制 K 线图,并应用自定义风格 mpf.plot(df, type='candle', style=s, title=f"{symbol} K 线图", ylabel='价格', ylabel_lower='成交量', volume=True, mav=(5,20,60), show_nontrading=False)# 调用函数,获取股票数据并绘制 K 线图get_stock_data_eastmoney('600519', '20200101', '20201231') # 示例股票代码和日期范围代码说明首先,导入所需的库: akshare: 用于获取东方财富网的股票数据 mplfinance: 用于绘制 K 线图 pandas: 用于数据处理 matplotlib: 用于图形显示然后,定义函数 get_stock_data_eastmoney,用于获取指定股票代码和日期范围的股票日线数据。在函数中,首先使用 akshare.stock_zh_a_hist 函数获取数据。接着,检查获取的数据是否为空。然后,将日期列设置为索引,并转换为 datetime 类型。之后,调整 DataFrame 列名以符合 mplfinance 的要求。接下来,定义 mplfinance 的自定义风格。最后,使用 mplfinance.plot 函数绘制 K 线图,并应用自定义风格。最后,调用函数,获取示例股票代码和日期范围的 K 线图。

数据获取与处理

akshare是一个开源财经数据接口库,它提供了获取各种财经数据的API,包括股票、期货、基金等。本文中,我们使用akshare的stock_zh_a_hist函数,从东方财富网抓取指定股票的历史日线数据。这个函数需要输入股票代码、数据周期(日线)、起始日期和结束日期。通过调整“qfq”参数,我们可以获取到前复权的数据,这对于长期分析尤为重要,因为它考虑了股票分红和拆分的影响。

获取到的数据是一个DataFrame,我们通过pandas库对其进行处理。首先,将日期列设置为索引并转换为datetime类型,这对后续的时间序列分析至关重要。接下来,我们将列名调整为mplfinance库能够识别的格式(小写的'open'、'close'、'high'、'low'、'volume'),以便进行K线图的绘制。

数据可视化

mplfinance是一个专门用于财经数据可视化的库,它提供了绘制K线图、成交量图和移动平均线等功能。在本文中,我们利用mplfinance绘制了包含K线、成交量和移动平均线(5天、20天、60天)的图表。为了使图表更加直观和美观,我们通过mpf.make_marketcolors和mpf.make_mpf_style自定义了图表的风格和颜色,其中红色代表上涨,绿色代表下跌。

为了支持中文显示并解决一些常见的显示问题(如负号'-'显示为方块),我们还对matplotlib的全局配置进行了调整,设置了中文字体和解决了负号显示问题。

结论

通过上述步骤,我们不仅获取了指定股票的历史数据,还将其以直观的形式展现了出来。这对于分析股票的历史表现、识别市场趋势和做出投资决策具有重要意义。Python和其相关库的强大功能,使得这一切变得既简单又高效。

0 阅读:2

余汉波看波涛

简介:感谢大家的关注