温馨提示×

如何使用Matplotlib绘制股票价格的蜡烛图或OHLC图

小亿
179
2024-05-21 09:38:10
栏目: 编程语言

要使用Matplotlib绘制股票价格的蜡烛图或OHLC图,你可以首先导入必要的库:

```python

import matplotlib.pyplot as plt

from mpl_finance import candlestick_ohlc

import pandas as pd

import matplotlib.dates as mdates

```

然后,你需要准备股票价格数据,通常是一个包含时间序列、开盘价、最高价、最低价和收盘价的DataFrame。确保时间序列是日期时间格式:

```python

data = pd.read_csv('stock_data.csv')

data['Date'] = pd.to_datetime(data['Date'])

data = data.set_index('Date')

```

接下来,你需要将数据转换成Matplotlib可识别的格式,并绘制蜡烛图或OHLC图:

```python

fig, ax = plt.subplots()

ohlc = data[['Open', 'High', 'Low', 'Close']].resample('D').ohlc()

ohlc.reset_index(inplace=True)

ohlc['Date'] = ohlc['Date'].map(mdates.date2num)

candlestick_ohlc(ax, ohlc.values, width=0.6, colorup='g', colordown='r')

ax.xaxis_date()

ax.xaxis.set_major_formatter(mdates.DateFormatter('%Y-%m-%d'))

ax.xaxis.set_major_locator(mdates.WeekdayLocator())

plt.title('Stock Price Candlestick Chart')

plt.xlabel('Date')

plt.ylabel('Price')

plt.show()

```

这样就可以绘制出股票价格的蜡烛图或OHLC图了。你可以根据需要对图表进行进一步的自定义和美化。

0