如何使用Matplotlib繪制股票價(jià)格的蠟燭圖或OHLC圖

小億
151
2024-05-21 09:38:10

要使用Matplotlib繪制股票價(jià)格的蠟燭圖或OHLC圖,你可以首先導(dǎo)入必要的庫(kù):

```python

import matplotlib.pyplot as plt

from mpl_finance import candlestick_ohlc

import pandas as pd

import matplotlib.dates as mdates

```

然后,你需要準(zhǔn)備股票價(jià)格數(shù)據(jù),通常是一個(gè)包含時(shí)間序列、開(kāi)盤價(jià)、最高價(jià)、最低價(jià)和收盤價(jià)的DataFrame。確保時(shí)間序列是日期時(shí)間格式:

```python

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

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

data = data.set_index('Date')

```

接下來(lái),你需要將數(shù)據(jù)轉(zhuǎn)換成Matplotlib可識(shí)別的格式,并繪制蠟燭圖或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()

```

這樣就可以繪制出股票價(jià)格的蠟燭圖或OHLC圖了。你可以根據(jù)需要對(duì)圖表進(jìn)行進(jìn)一步的自定義和美化。

0