張郎生活的筆記

今天就來個簡單的python的talib紀錄

回顧上一次的, 把中鋼的資料抓到今天

from pandas_datareader import data as web

import pandas as pd

import datetime as dt

import fix_yahoo_finance as yf

yf.pdr_override()

start = dt.datetime(2000, 1, 1)

end = dt.datetime(2018, 8, 28)

df = web.get_data_yahoo(['2002.TW'],start, end)

df.to_csv(r'c:\2002.TW.csv')

df = pd.read_csv(r'c:\2002.tw.csv')

df.head()

那我們需要把Date這一個column作成index(inplace=True表示不會生成新的dataframe)

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

df.set_index("Date", inplace=True)

df.head()

看一下Date的index類型

df.index

這時候這個index已經轉成DatetimeIndex, 是我們所需要的類型

然後來試畫看看收盤價的線圖

import matplotlib.pyplot as plt

df['Close'].plot()

plt.show()

然後

from talib import abstract

abstract.SMA(df).plot()

咦? 意思是Close要改成小寫嗎?

df.rename(columns = {'Open':'open', 'High':'high', 'Low':'low', 'Close':'close'}, inplace = True)

SMA = abstract.SMA(df).plot()

plt.show(SMA)

OK~大家可以試看看~