pandas提供了一套標准的時間序列處理工具和算法,使得我們可以非常高效的處理時間序列,比如切片、聚合、重采樣等等。
本節我們討論以下三種時間序列:
- 時間戳(time stamp):一個特定的時間點,比如2018年4月2日。
- 時間間隔和時期(time period):兩個特定的時間起點和終點,比如2018年4月2日-2019年4月2日。
- 時間差或持續時間(time deltas):特定的時間長度,比如20秒。
Python的時間序列處理是一個很重要的話題,尤其在金融領域有着非常重要的應用。本節只做簡單的介紹,如果以后有需要的話再深入學習。本文參考的Working with Time Series對python原生的日期時間處理模塊datetime
,numpy日期時間處理模塊datetime64
,以及第三方日期時間處理模塊dateutil
等都做了介紹,我們這里跳過這些,直接進入pandas的時間序列。
import numpy as np
import pandas as pd
1. pandas時間序列:時間索引
當你使用時間戳作為數據索引的時候,pandas的時間序列工具就變得是非有用。
index = pd.DatetimeIndex(['2014-07-04', '2014-08-04',
'2015-07-04', '2015-08-04'])
data = pd.Series([0, 1, 2, 3], index=index)
data
#> 2014-07-04 0
#> 2014-08-04 1
#> 2015-07-04 2
#> 2015-08-04 3
#> dtype: int64
data['2014-07-04':'2015-07-04']
#> 2014-07-04 0
#> 2014-08-04 1
#> 2015-07-04 2
#> dtype: int64
data['2015']
#> 2015-07-04 2
#> 2015-08-04 3
#> dtype: int64
2. pandas時間序列數據結構
-
對於時間戳,pandas提供了
Timestamp
類型。它基於numpy.datetime64
數據類型,與與之相關的索引類型是DatetimeIndex
。 -
對於時間時期,pandas提供了
Period
類型,它是基於numpy.datetime64
編碼的固定頻率間隔。與之相關的索引類型是PeriodIndex
。 -
對於時間差,pandas提供了
Timedelta
類型,同樣基於numpy.datetime64
。與之相關的索引結構是TimedeltaIndex
。
最基本的時間/日期是時間戳Timestamp
和DatetimeIndex
。但是我們可以直接使用pd.to_datetime()
函數來直接解析不同的格式。傳入單個日期給pd.to_datetime()
返回一個Timestamp
,傳入一系列日期默認返回DatetimeIndex
。
dates = pd.to_datetime([datetime(2015, 7, 3), '4th of July, 2015',
'2015-Jul-6', '07-07-2015', '20150708'])
dates
#> DatetimeIndex(['2015-07-03', '2015-07-04', '2015-07-06', '2015-07-07',
#> '2015-07-08'],
#> dtype='datetime64[ns]', freq=None)
當一個日期減去另一個日期的時候,TimedeltaIndex
就會被創建:
dates - dates[0]
#> TimedeltaIndex(['0 days', '1 days', '3 days', '4 days', '5 days'], dtype='timedelta64[ns]', freq=None)
2.1 定期序列
為了方便地創建定期的日期序列,pandas提供了一些函數:pd.date_range()
創建時間戳,pd.period_range()
創建時期,pd.timedelta_range()
創建時間差。默認情況下頻率間隔是一天。
pd.date_range('2015-07-03', '2015-07-10')
#> DatetimeIndex(['2015-07-03', '2015-07-04', '2015-07-05', '2015-07-06',
#> '2015-07-07', '2015-07-08', '2015-07-09', '2015-07-10'],
#> dtype='datetime64[ns]', freq='D')
另外可以不指定終點,而是提供一個時期數:
pd.date_range('2015-07-03', periods=8)
#> DatetimeIndex(['2015-07-03', '2015-07-04', '2015-07-05', '2015-07-06',
#> '2015-07-07', '2015-07-08', '2015-07-09', '2015-07-10'],
#> dtype='datetime64[ns]', freq='D')
頻率間隔可以通過freq
參數設置,默認是D
:
pd.date_range('2015-07-03', periods=8, freq='H')
#> DatetimeIndex(['2015-07-03 00:00:00', '2015-07-03 01:00:00',
#> '2015-07-03 02:00:00', '2015-07-03 03:00:00',
#> '2015-07-03 04:00:00', '2015-07-03 05:00:00',
#> '2015-07-03 06:00:00', '2015-07-03 07:00:00'],
#> dtype='datetime64[ns]', freq='H')
類似的
pd.period_range('2015-07', periods=8, freq='M')
#> PeriodIndex(['2015-07', '2015-08', '2015-09', '2015-10', '2015-11', '2015-12',
#> '2016-01', '2016-02'],
#> dtype='period[M]', freq='M')
pd.timedelta_range(0, periods=10, freq='H')
#> TimedeltaIndex(['00:00:00', '01:00:00', '02:00:00', '03:00:00', '04:00:00',
#> '05:00:00', '06:00:00', '07:00:00', '08:00:00', '09:00:00'],
#> dtype='timedelta64[ns]', freq='H')
3. 頻率和偏移
以上這些時間序列處理工具最基礎的概念是頻率和時間偏移。下表總結了主要的代碼表示:
代碼 | 描述 | 代碼 | 描述 |
---|---|---|---|
D |
日歷天 | L |
厘秒 |
W |
星期 | U |
毫秒 |
M |
月末 | N |
納秒 |
Q |
季度末 | B |
工作日 |
A |
年末 | BM |
工作月末 |
H |
小時 | BQ |
工作季末 |
T |
分鍾 | BA |
工作年末 |
S |
秒 | BH |
工作時 |
上面的月,季度以及周期性頻率都被標記為這一時期的終點,如果想變成起始點可以加S
后綴。
代碼 | 描述 | 代碼 | 描述 |
---|---|---|---|
MS |
月初 | BMS |
工作月初 |
QS |
季度初 | BQS |
工作季初 |
AS |
年初 | BAS |
工作年初 |
另外,還可以通過加后綴的方法改變季度或年的月份等。
Q-JAN
,BQ-FEB
,BQS-APR
等等;-
A-JAN
,BA-FEB
,BAS-APR
等等。類似的,星期也可以分解成周一,周二…
-
W-SUN
,W-MON
,W-WED
等等以上這些代碼還可以與數字結合來表示頻率, 例如我們要表示2小時30分的頻率:
pd.timedelta_range(0, periods=9, freq="2H30T")
#> TimedeltaIndex(['00:00:00', '02:30:00', '05:00:00', '07:30:00', '10:00:00',
#> '12:30:00', '15:00:00', '17:30:00', '20:00:00'],
#> dtype='timedelta64[ns]', freq='150T')
以上這些代碼表示的含義還可以通過pd.tseries.offsets
模塊表示:
from pandas.tseries.offsets import BDay
pd.date_range('2015-07-01', periods=5, freq=BDay())
#> DatetimeIndex(['2015-07-01', '2015-07-02', '2015-07-03', '2015-07-06',
#> '2015-07-07'],
#> dtype='datetime64[ns]', freq='B')
更加詳細的討論可以看DateOffset。
4. 重采樣,轉移,加窗口
我們以一些真實的金融數據為例講解,pandas-datareader
(可通過conda install pandas-datareader
來安裝, 如果)包包含谷歌,雅虎等公司的金融數據。
from pandas_datareader import data
goog = data.DataReader('GOOG', start='2004', end='2016',
data_source='google')
goog.head()
#> Open High Low Close Volume
#> Date
#> 2004-08-19 49.96 51.98 47.93 50.12 NaN
#> 2004-08-20 50.69 54.49 50.20 54.10 NaN
#> 2004-08-23 55.32 56.68 54.47 54.65 NaN
#> 2004-08-24 55.56 55.74 51.73 52.38 NaN
#> 2004-08-25 52.43 53.95 51.89 52.95 NaN
簡單起見,我們只是用收盤價格
goog = goog['Close']
為了更好的說明問題,我們可以利用matplotlib對數據進行可視化,關於matplotlib可以通過Python for Data Analysis這本書進行學習,也可以通過官網的gallary進行學習,我個人當時是在官網看了大量的實例,有什么需求直接去官網找例子,都很方便。本系列學習筆記不會包含數據可視化部分。
import matplotlib.pyplot as plt
goog.plot()
4.1 重采樣及頻率轉換
重采樣(resampling)指的是將時間序列從一個頻率轉換到另一個頻率的處理過程。將高頻數據聚合到低頻稱為降采樣(downsampling),將低頻數據轉換到高頻則稱為升采樣(upsampling)。除此以外還存在一種采樣方式既不是升采樣,也不是降采樣,比如W-WED
轉換成W-FRI
。
可以通過resample()
函數來實現,也可以通過更簡單的方式asfreq()
函數來實現。兩者基本的不同點在於resample()
是一種數據聚合方式asfreq()
是一種數據選取方式。
goog.plot(alpha=0.5, style='-')
goog.resample('BA').mean().plot(style=':')
goog.asfreq('BA').plot(style='--');
plt.legend(['input', 'resample', 'asfreq'],
loc='upper left')
resample()
的處理過程是取整年的數據的平均值,而asfreq()
是選取年末的時間點的數據。
對於升采樣來說,resample()
與asfreq()
是等效的。對於空置采樣點都會用NaN
來進行填充。就像pd.fillna()
方法,asfreq()
接受一個參數method
可指定缺失值的填充方法。ffill
表示與前面的值保持一致,bfill
表示與后面的值保持一致等等。
fig, ax = plt.subplots(2, sharex=True)
data = goog.iloc[:10]
data.asfreq('D').plot(ax=ax[0], marker='o')
data.asfreq('D', method='bfill').plot(ax=ax[1], style='-o')
data.asfreq('D', method='ffill').plot(ax=ax[1], style='--o')
ax[1].legend(["back-fill", "forward-fill"])
4.2 時間移動
另一個與時間序列相關的操作是數據在時間軸的移動。pandas提供了兩個相關的函數:shift()
和tshift()
,兩者的不同是shift()
移動的是數據,tshift()
移動的是時間軸(索引)。
fig, ax = plt.subplots(3, sharey=True)
# apply a frequency to the data
goog = goog.asfreq('D', method='pad')
goog.plot(ax=ax[0])
goog.shift(900).plot(ax=ax[1])
goog.tshift(900).plot(ax=ax[2])
# legends and annotations
local_max = pd.to_datetime('2007-11-05')
offset = pd.Timedelta(900, 'D')
ax[0].legend(['input'], loc=2)
ax[0].get_xticklabels()[2].set(weight='heavy', color='red')
ax[0].axvline(local_max, alpha=0.3, color='red')
ax[1].legend(['shift(900)'], loc=2)
ax[1].get_xticklabels()[2].set(weight='heavy', color='red')
ax[1].axvline(local_max + offset, alpha=0.3, color='red')
ax[2].legend(['tshift(900)'], loc=2)
ax[2].get_xticklabels()[1].set(weight='heavy', color='red')
ax[2].axvline(local_max + offset, alpha=0.3, color='red')
時移一個重要作用即使計算一段時間內的差別,比如計算谷歌股票一年的投資回報率:
ROI = 100 * (goog.tshift(-365) / goog - 1)
ROI.plot()
plt.ylabel('% Return on Investment')
4.3 滾動窗口
滾動統計是時間序列的又一重要的操作。可以通過Series
和DataFrame
的rolling()
方法實現,返回類似於groupby
的操作,同樣也有許多數據聚合的操作。
rolling = goog.rolling(365, center=True)
data = pd.DataFrame({'input': goog,
'one-year rolling_mean': rolling.mean(),
'one-year rolling_std': rolling.std()})
ax = data.plot(style=['-', '--', ':'])
ax.lines[0].set_alpha(0.3)