将PANDAS数据帧从每月转换为每天

2024-04-29 05:09:06 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个数据框架,其中包含317只股票的2014年月度数据(317只股票x 12个月=DF中的3804行)。我想将其转换为每日数据帧(317个tickers x 365天=115705行)。所以,我认为我需要增加样本或重新编制索引,同时将每月价值分散到每月的每一天,但我无法让它正常工作。

数据帧当前采用以下格式:

>>> df
month    ticker   b    c
2014-1   AAU      10   .04     #different values every month for each ticker
2014-2   AAU      20   .03
2014-3   AAU      13   .06
.
2014-12  AAU      11   .03
.
.
.
2014-1   ZZY      11   .11
2014-2   ZZY      6    .03
.
2014-12  ZZY      17   .09

这就是我想要的:

>>> df
day          ticker   b    c
2014-01-01   AAU      10   .04  #same values every day in month for each ticker
2014-01-02   AAU      10   .04
2014-01-03   AAU      10   .04
.
2014-01-31   AAU      10   .04
2014-02-01   AAU      20   .03
2014-02-02   AAU      20   .03
.
2014-02-28   AAU      20   .03
.
.
.
2014-12-30   ZZY      17   .09 
2014-12-31   ZZY      17   .09 

我尝试过将groupby与按天重新采样相结合,但更新后的数据帧将以日期“2014-01-13”开始,而不是1月1日,以日期“2014-12-01”结束,而不是12月31日。我还尝试将月份值从例如“2014-1”更改为“2014-01-01”等。,但重新采样的数据帧仍在“2014-01-01”结束。必须有一个更简单的方法来解决这个问题,所以我很感谢你的帮助。我整天都在绕圈子。


Tags: 数据框架dffor股票tickereachvalues
2条回答

让我们做一个合成实验。假设我们有每日时间序列数据:

dates = pd.date_range(start, end, freq='D')
ts = pd.Series(data, index=dates)

通过平均一个月内的所有数据生成每月时间序列:

ts_mon = ts.resample('MS', how='mean')

现在尝试将这个月时间序列重新采样到日时间序列,在一个月内使用统一的值。使用reindex从@unutbu借用一个步骤的第一个方法工作得很好:

ts_daily = ts_mon.reindex(dates, method='ffill')
Out:
  2000-01-01 100.21
  2000-01-02 100.21
  ...
  2000-12-30 80.75
  2000-12-31 80.75

使用resample的第二个方法不起作用,因为它返回上个月的第一天:

ts_daily = ts_mon.resample('D').ffill()
Out:
  2000-01-01 100.21
  2000-01-02 100.21
  ...
  2000-11-30 99.33
  2000-12-01 80.75

首先,将月日期字符串解析为Pandas时间戳:

df['month'] = pd.to_datetime(df['month'], format='%Y-%m')
#        month ticker   b     c
# 0 2014-01-01    AAU  10  0.04
# 1 2014-02-01    AAU  20  0.03
# 2 2014-03-01    AAU  13  0.06
# 3 2014-12-01    AAU  11  0.03
# 4 2014-01-01    ZZY  11  0.11
# 5 2014-02-01    ZZY   6  0.03
# 6 2014-12-01    ZZY  17  0.09

接下来,以月份作为索引,以ticker作为列级别,对DataFrame进行透视:

df = df.pivot(index='month', columns='ticker')
#              b         c      
# ticker     AAU ZZY   AAU   ZZY
# month                         
# 2014-01-01  10  11  0.04  0.11
# 2014-02-01  20   6  0.03  0.03
# 2014-03-01  13 NaN  0.06   NaN
# 2014-12-01  11  17  0.03  0.09

通过现在旋转,我们以后可以更容易地向前填充每一列。

现在查找开始日期和结束日期:

start_date = df.index.min() - pd.DateOffset(day=1)
end_date = df.index.max() + pd.DateOffset(day=31)

有趣的是,添加pd.DateOffset(day=31)并不总是导致日期在第31天结束。如果月份是二月,则添加pd.DateOffset(day=31)返回二月的最后一天:

In [130]: pd.Timestamp('2014-2-28') + pd.DateOffset(day=31)
Out[130]: Timestamp('2014-02-28 00:00:00')

这很好,因为这意味着添加pd.DateOffset(day=31)将始终给我们一个月中最后一个有效日。

现在我们可以重新索引并向前填充数据帧:

dates = pd.date_range(start_date, end_date, freq='D')
dates.name = 'date'
df = df.reindex(dates, method='ffill')

会产生

In [160]: df.head()
Out[160]: 
             b         c      
ticker     AAU ZZY   AAU   ZZY
date                          
2014-01-01  10  11  0.04  0.11
2014-01-02  10  11  0.04  0.11
2014-01-03  10  11  0.04  0.11
2014-01-04  10  11  0.04  0.11
2014-01-05  10  11  0.04  0.11

In [161]: df.tail()
Out[161]: 
             b         c      
ticker     AAU ZZY   AAU   ZZY
date                          
2014-12-27  11  17  0.03  0.09
2014-12-28  11  17  0.03  0.09
2014-12-29  11  17  0.03  0.09
2014-12-30  11  17  0.03  0.09
2014-12-31  11  17  0.03  0.09

要将ticker移出列索引并移回列中,请执行以下操作:

df = df.stack('ticker')
df = df.sortlevel(level=1)
df = df.reset_index()

所以把这些放在一起:

import pandas as pd
df = pd.read_table('data', sep='\s+')
df['month'] = pd.to_datetime(df['month'], format='%Y-%m')
df = df.pivot(index='month', columns='ticker')

start_date = df.index.min() - pd.DateOffset(day=1)
end_date = df.index.max() + pd.DateOffset(day=31)
dates = pd.date_range(start_date, end_date, freq='D')
dates.name = 'date'
df = df.reindex(dates, method='ffill')

df = df.stack('ticker')
df = df.sortlevel(level=1)
df = df.reset_index()

收益率

In [163]: df.head()
Out[163]: 
        date ticker   b     c
0 2014-01-01    AAU  10  0.04
1 2014-01-02    AAU  10  0.04
2 2014-01-03    AAU  10  0.04
3 2014-01-04    AAU  10  0.04
4 2014-01-05    AAU  10  0.04

In [164]: df.tail()
Out[164]: 
          date ticker   b     c
450 2014-12-27    ZZY  17  0.09
451 2014-12-28    ZZY  17  0.09
452 2014-12-29    ZZY  17  0.09
453 2014-12-30    ZZY  17  0.09
454 2014-12-31    ZZY  17  0.09

相关问题 更多 >