我正在尝试按日期聚合Pandas DataFrame,然后按日期索引到结果中,但我似乎根本无法对其进行索引。
from datetime import date, datetime
import pandas as pd
import numpy as np
ts = pd.DataFrame({'data': np.random.randn(60)}, index=pd.date_range('1/1/2000', periods=60, freq='2h'))
g = ts.groupby(ts.index.date).agg(np.mean)
print(g)
data
2000-01-01 0.090038
2000-01-02 0.099970
2000-01-03 -0.619274
2000-01-04 0.027040
2000-01-05 -0.323205
首先,我希望索引是DatetimeIndex
,但是:
print(g.index)
Index([2000-01-01, 2000-01-02, 2000-01-03, 2000-01-04, 2000-01-05], dtype='object')
g.index[0]
datetime.date(2000, 1, 1)
好的,让我们尝试索引它:
g['2000-01-01']
KeyError: u'no item named 2000-01-01'
g[date(2000, 01, 01)]
KeyError: u'no item named 2000-01-01'
g[datetime(2000, 01, 01)]
KeyError: u'no item named 2000-01-01 00:00:00'
g[pd.to_datetime('2000-01-01')]
KeyError: u'no item named 2000-01-01 00:00:00'
我甚至无法用索引本身索引!
g[g.index[0]]
KeyError: u'no item named 2000-01-01'
我错过了什么?为什么不是DatetimeIndex
一个DatetimeIndex
或至少一个PeriodIndex
的分组/聚合结果?有没有办法让它成为一个?
答案 0 :(得分:2)
索引是一个日期时间,因为您通过一组日期时间传递group:
In [11]: ts.index.date[:2]
Out[11]: array([datetime.date(2000, 1, 1), datetime.date(2000, 1, 1)], dtype=object)
更常用的方法是重新采样:
In [12]: res = ts.resample('D', how='mean')
In [13]: res
Out[13]:
data
2000-01-01 0.181246
2000-01-02 -0.167023
2000-01-03 -0.075843
2000-01-04 -0.218141
2000-01-05 -0.144635
或者如果您正在做一些更复杂的事情,可以使用TimeGrouper:
In [14]: ts.groupby(pd.TimeGrouper('D')).mean()
Out[14]:
data
2000-01-01 0.181246
2000-01-02 -0.167023
2000-01-03 -0.075843
2000-01-04 -0.218141
2000-01-05 -0.144635
这些索引是DatetimeIndex。
关于访问行,你应该使用loc,虽然我喜欢使用时间戳访问:
In [15]: res.loc['2000-01-01'] # KeyError without using loc
Out[15]:
data 0.181246
Name: 2000-01-01 00:00:00, dtype: float64
In [16]: res.loc[pd.Timestamp('2000-01-01')]
Out[16]:
data 0.181246
Name: 2000-01-01 00:00:00, dtype: float64
你可以使用这些字符串切片(但我不认为它是一个超级健壮的主意,我更喜欢使用时间戳):
In [17]: res['2000-01-01':'2000-01-01']
Out[17]:
data
2000-01-01 0.181246