索引按日期分组并聚合的Pandas数据框?

时间:2014-04-03 16:41:09

标签: python pandas

我正在尝试按日期聚合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的分组/聚合结果?有没有办法让它成为一个?

1 个答案:

答案 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