家庭作業-1 @ richkan的部落格

今天第一個家庭作業: 上述列出的指標，任選三個進行指標計算與繪圖(指標圖)
作業要求:
- 列出程式碼
- 印出結果與截圖，放入WORD中
- 可以有許多寫法，同學可以自行決定寫法，創新出自己的程式碼或繪圖方式(例如把移動平均線SMA放在K線圖中是最棒的寫法，大家可以參考mlpfinance套件與talib結合起來

from pandas_datareader import data
import matplotlib.pyplot as plt
import pandas as pd
import numpy as np
import talib

plt.rcParams['figure.figsize'] = [10, 5]
plt.rcParams['figure.dpi'] = 120

# 讀取 2020-01-01 到最新日期
data = data.DataReader("^TWII", "yahoo", "2020-01-01")

# 收盤價
close = data['Close']
close.plot(label='Close', legend=True)

# 5 日均線
SMA5 = talib.SMA(close, timeperiod=5)
SMA5.plot(label='SMA5', legend=True)

# 20日均線
SMA20 = talib.SMA(close, timeperiod=20)
SMA20.plot(label='SMA20', legend=True)

# 顯示標題
plt.title('^TWII')
plt.savefig('^TWII.png')
plt.show()

# 強弱指標RSI
RSI5 = talib.RSI(close, timeperiod=5)
RSI5['2021'].plot(label='RSI5', legend=True)

# KD 指標
data['k'], data['d'] = talib.STOCH(data['High'], data['Low'], data['Close'])
K=data['k']
K['2021'].plot(label='K', legend=True)

D=data['d']
D['2021'].plot(label='D', legend=True)

# WILLR 指標
data['WILLR5'] = talib.WILLR(data['High'], data['Low'], data['Close'], timeperiod=5)
WILLR5 = data['WILLR5'] 
WILLR5['2021'].plot(label='WILLR5', legend=True)

# 顯示標題
plt.title('^TWII')
plt.savefig('^TWII-RSI.png')
plt.show()