Pandas - 将数据帧多索引转换为datetime对象

时间:2014-06-06 21:14:28

标签: python pandas

考虑输入文件b.dat

string,date,number
a string,2/5/11 9:16am,1.0
a string,3/5/11 10:44pm,2.0
a string,4/22/11 12:07pm,3.0
a string,4/22/11 12:10pm,4.0
a string,4/29/11 11:59am,1.0
a string,5/2/11 1:41pm,2.0
a string,5/2/11 2:02pm,3.0
a string,5/2/11 2:56pm,4.0
a string,5/2/11 3:00pm,5.0
a string,5/2/14 3:02pm,6.0
a string,5/2/14 3:18pm,7.0

我可以像这样分组每月总数:

b=pd.read_csv('b.dat')
b['date']=pd.to_datetime(b['date'],format='%m/%d/%y %I:%M%p')
b.index=b['date']
bg=pd.groupby(b,by=[b.index.year,b.index.month])
bgs=bg.sum()

分组总计的索引如下所示:

bgs

            number
2011 2       1
     3       2
     4       8
     5      14
2014 5      13

bgs.index

MultiIndex(levels=[[2011, 2014], [2, 3, 4, 5]],
       labels=[[0, 0, 0, 0, 1], [0, 1, 2, 3, 3]])

我想将索引重新格式化为日期时间格式(天数可以是月份的第一天)。

我尝试过以下方法:

bgs.index = pd.to_datetime(bgs.index)

bgs.index = pd.DatetimeIndex(bgs.index)

两者都失败了。有谁知道我怎么能这样做?

2 个答案:

答案 0 :(得分:5)

按'M'考虑resample,而不是按DatetimeIndex的属性进行分组:

In [11]: b.resample('M', how='sum').dropna()
Out[11]:
            number
date
2011-02-28       1
2011-03-31       2
2011-04-30       8
2011-05-31      14
2014-05-31      13

注意:如果你不想要几个月,你必须放弃NaN。

答案 1 :(得分:4)

您可以通过所需的日期计算从索引创建列,然后将其设置为索引:

bgs['expanded_date'] = bgs.index.map(lambda x: datetime.date(x.year, x.month, 1))
bgs.set_index('expanded_date')