pandas.jpg

首先，強烈推薦關注Dr. Fish, 每篇文章都很有深度。因為受到她《用Python淺析股票數據》文章的啟發，所以分享下最近幾天學習獲取股票交易歷史數據的總結.

pandas 官方文檔
matplotlib 官方文檔
上證指數

首先，需要引入相應的包

import pandas as pd
import numpy as np
from pandas_datareader import data, wb # 需要安裝 pip install pandas_datareader
import datetime
import matplotlib.pyplot as plt
import matplotlib
matplotlib.style.use('ggplot')
%matplotlib inline
%pylab inline
%config InlineBackend.figure_format = 'retina'

如果采用anaconda，安裝辦法參考: 檢索和安裝package

如果是pycharm等其他開發環境，請

pip install pandas_datareader 

利用DataReader獲取數據,今天是20170420，獲取從4月1日到19日的數據

# 定義獲取數據的時間段
start = datetime.datetime(2017, 4, 1)
end = datetime.date.today()

# 獲取股票信息 ex: 中國石油
# 如果要看上證指數請參考換成600000.ss
# 如果要看深成指請換成000001.sz
cnpc = data.DataReader("601857.SS", 'yahoo', start, end)

cnpc.head(5)

獲取統計信息

cnpc.describe() # 數據整體概覽

#修改索引和列的名稱
cnpc.rename(columns={'Open':'open', 'High':'high', 'Low':'low', 'Close':'close','Volume':'volumne','Adj Close':'adj close'}, inplace=True)
cnpc.index.rename('date', inplace=True)

因為實驗數據太少，現將20170401 改成20170301

cnpc.columns
# outputs: Index(['open', 'high', 'low', 'close', 'volumne', 'adj close', 'change','pct_change'],dtype='object')

試做收盤價曲線

cnpc['close'].plot(grid=True)

收盤曲線圖

我們看到由于清明節放假原因4月3號，4號休市。剔除volume為空的行

cnpc[cnpc.volume != 0]

計算漲跌值

# 利用 diff 函數快速計算漲跌幅
change=cnpc.close.diff()
#插入列
cnpc.insert(6,'change', change)
cnpc.tail(5)

||open|high|low|close|volumne|adj close|change|
| ------------- |:-------------:| -----:|-----:|-----:|-----:|-----:|
|date||||||||
|2017-04-13|8.08|8.11|8.03|8.06|29579200|8.06|-0.04|
|2017-04-14|8.05|8.07|8.00|8.02|27295800|8.02|-0.04|
|2017-04-17|7.99|8.07|7.95|8.05|34244900|8.05|0.03|
|2017-04-18|8.02|8.05|7.95|7.96|25706900|7.96|-0.09|
|2017-04-19|7.90|7.92|7.75|7.80|37354400|7.80|-0.16 |

計算漲跌幅

# 用shift方法錯位
# cnpc['pct_change'] = ((cnpc['Change'] - sh['Change'].shift(1)) / sh['Change'])
# 或用pct_Change函數
cnpc.change.pct_change()
cnpc.insert(7,'pct_change', cnpc.change.pct_change())
cnpc.tail(5)

||open|high|low|close|volume|adj close|change|pct_change|
| ------------- |:-------------:| -----:|-----:|-----:|-----:|-----:|-----:|
|date||||||||
|2017-04-13|8.08|8.11|8.03|8.06|29579200|8.06|-0.04|-1.500000e+00|
|2017-04-14|8.05|8.07|8.00|8.02|27295800|8.02|-0.04|4.440892e-14|
|2017-04-17|7.99|8.07|7.95|8.05|34244900|8.05|0.03|-1.750000e+00|
|2017-04-18|8.02|8.05|7.95|7.96|25706900|7.96|-0.09|-4.000000e+00|
|2017-04-19|7.90|7.92|7.75|7.80|37354400|7.80|-0.16|7.777778e-01|

shift的用法
shift函數是對數據進行移動的操作，假如現在有一個DataFrame數據df

| index | value1 |
| ------------- | -----:||
| A| 0 |
| B| 1 |
| C | 2 |
| D | 3 |

df.shift()

| index | value1 |
| ------------- | -----:||
| A| NaN |
| B| 0|
| C | 1 |
| D | 2 |
函數原型：

DataFrame.shift(periods=1, freq=None, axis=0)
#periods：類型為int，表示移動的幅度，可以是正數，也可以是負數，默認值是1,1就表示移動一次，注意這里移動的都是數據，而索引是不移動的，移動之后沒有對應值的，就賦值為NaN。
執行以下代碼：
df.shift(2)

| index | value1 |
| ------------- | -----:||
| A| NaN |
| B| NaN|
| C | 0|
| D | 1 |

df.shift(-1)

| index | value1 |
| ------------- | -----:||
| A| 1|
| B| 2|
| C | 3|
| D | NaN |

計算5日，20日均線

cnpc["ma5"] = np.round(cnpc["close"].rolling(window = 5, center = False).mean(), 2)
cnpc["ma20"] = np.round(cnpc["close"].rolling(window = 20, center = False).mean(), 2)

5日均線.png

用matplotlib圖形化顯示價格信息以及變動

rom datetime import datetime
from dateutil.parser import parse
from matplotlib.dates import AutoDateLocator, DateFormatter,DayLocator


# date = cnpc['date']
# date = pd.to_datetime(date)
date = cnpc.index
high = cnpc['high'].values
low = cnpc['low'].values
open= cnpc['open'].values
close = cnpc['close'].values
ma5 = cnpc['ma5'].values
ma10 = cnpc['ma10'].values
ma20 = cnpc['ma20'].values
# turnover=stock_data['turnover'].values


fig = plt.figure(figsize = (15,15))
ax = fig.add_subplot(211)
ax.set_title("Stock price")
ax.plot(date,open,label='open')
ax.plot(date,high,label='high')
ax.plot(date,low,label = 'low')
ax.plot(date,close,label = 'close')
ax.plot(date,ma5,label = 'ma5')
ax.plot(date,ma10,label = 'ma10')
ax.plot(date,ma20,label = 'ma20')
# ax.plot(date,turnover,label='turnover')

ax.set_xlabel("date")
ax.set_ylabel("values")
ax.xaxis.set_major_locator(DayLocator(bymonthday=range(1,32), interval=1)) 
ax.xaxis.set_major_formatter(DateFormatter('%Y%m%d')) 
plt.xticks(rotation=60)
plt.legend(loc='upper left') 

plt.grid(True)
plt.show()

價格變動.png

假設5日均線與20日均線的交叉點，是交易的時機。移動平均線策略，最簡單的方式就是：當5日均線從下方超越20日均線時，買入股票，當5日均線從上方越到20日均線之下時，賣出股票。

為了找出交易的時機，我們計算5日均價和20日均價的差值，并取其正負號，作于下圖。當圖中水平線出現跳躍的時候就是交易時機。

cnpc['ma5-20'] = cnpc['ma5'] - cnpc['ma20']
cnpc['Diff'] = np.sign(cnpc['ma5-20']) # sign是取+-
cnpc['Diff'].dropna().plot(ylim=(-2,2)).axhline(y=0, color='black', lw=2)

Paste_Image.png

K線圖

首先非常感謝DrFish提供的思路，K線圖可以將最高價、最低價、開盤價、收盤價四個價格指標很好的顯示出來

Paste_Image.png

如下代碼引用自DrFish的文章：

from matplotlib.finance import candlestick_ohlc
from matplotlib.dates import DateFormatter, WeekdayLocator, DayLocator, MONDAY
# TUESDAY 僅僅在X粥主軸上顯示周二的日期
def pandas_candlestick_ohlc(stock_data, otherseries=None):    

    # 設置繪圖參數，主要是坐標軸 
    mondays = WeekdayLocator(MONDAY) 
    alldays = DayLocator()   
    dayFormatter = DateFormatter('%d')

    fig, ax = plt.subplots()
    fig.subplots_adjust(bottom=0.2)
    print(stock_data.index)
    print(stock_data.index[-1]) 
    print(stock_data.index[0]) 
    if stock_data.index[-1] - stock_data.index[0] < pd.Timedelta('730 days'):
        weekFormatter = DateFormatter('%b %d')  
        ax.xaxis.set_major_locator(mondays)
        ax.xaxis.set_minor_locator(alldays)
    else:
        weekFormatter = DateFormatter('%b %d, %Y')
    ax.xaxis.set_major_formatter(weekFormatter)
    ax.grid(True)

    # 創建K線圖   
    stock_array = np.array(stock_data.reset_index()[['date','open','high','low','close']])
    stock_array[:,0] = date2num(stock_array[:,0])
    candlestick_ohlc(ax, stock_array, colorup = "red", colordown="green", width=0.4)


    # 可同時繪制其他折線圖
    if otherseries is not None:
        for each in otherseries:
            plt.plot(stock_data[each], label=each)            
        plt.legend()


    ax.xaxis_date()
    ax.autoscale_view()
    plt.setp(plt.gca().get_xticklabels(), rotation=45, horizontalalignment='right')

    plt.show()

K線圖.png

相關系數展示

前一篇文章中，我只是通過循環print出各自的相關系數，但是沒有進行圖形化的展示，通過DrFish的講解, 此處引用當做筆記

part=cnpc[['open', 'high', 'low', 'close', 'volumne', 'adj close']]
cov = np.corrcoef(part.T) # 通過矩陣的轉置
cov
#array([[ 1.        ,  0.93329891,  0.91815383,  0.81640801, -0.12577619,
         0.81640801],
       [ 0.93329891,  1.        ,  0.938317  ,  0.92304648, -0.00264197,
         0.92304648],
       [ 0.91815383,  0.938317  ,  1.        ,  0.9365154 , -0.21158285,
         0.9365154 ],
       [ 0.81640801,  0.92304648,  0.9365154 ,  1.        , -0.10726339,
         1.        ],
       [-0.12577619, -0.00264197, -0.21158285, -0.10726339,  1.        ,
        -0.10726339],
       [ 0.81640801,  0.92304648,  0.9365154 ,  1.        , -0.10726339,
         1.        ]])

如果覺得看數字還是不夠方便，我們繼續將上述相關性矩陣轉換成圖形，如下圖所示，其中用顏色來代表相關系數。

img = plt.matshow(cov,cmap=plt.cm.winter)
plt.colorbar(img, ticks=[-1,0,1])
plt.show()

表示相關系數的散點圖.png

本文主要是為了熟悉pandas, matplotlib使用，供自己和大家學習參考.