2016-11-22 39 views
3

考虑以下numpy的数组:产生-1周围numpy的给定日期/1天

import numpy as np 
t = np.array(['2016-07-28', '2016-08-25', '2016-09-29', '2016-10-27', '2016-11-17'], dtype='datetime64[D]') 

怎样才能前一天和第二天每个数组元素?我所需的输出是如下:

array(['2016-07-27', '2016-07-28', '2016-07-29', ..., '2016-11-16', 
    '2016-11-17', '2016-11-18'], dtype='datetime64[D]') 

我目前的方法是,分别生成三个数组(以前,日,NEXT)和除以下将它们合并:

tt = np.sort(np.append(t1,(t,t2))) 
+0

你可以分享如何创建't1'和't2' -1/+ 1日期? – burhan

+0

t2 =(t + pd.Timedelta(days = 1))。astype('datetime64 [D]'); t1 =(t - pd.Timedelta(days = 1))。astype('datetime64 [D]')。 T创建的是pd.Series对象,但我最终将其作为一个numpy数组存储。希望这是有道理的。 –

回答

6

Broadcast这里呢!

(t[:,None] + np.arange(-1,2)).ravel() 

紧凑版本 -

(t[:,None] + [-1,0,1]).ravel() 

采样运行 -

In [191]: t 
Out[191]: array(['2016-07-28', '2016-08-25', '2016-09-29', '2016-10-27', '2016-11-17'], dtype='datetime64[D]') 

In [192]: (t[:,None] + np.arange(-1,2)).ravel() 
Out[192]: 
array(['2016-07-27', '2016-07-28', '2016-07-29', '2016-08-24', 
     '2016-08-25', '2016-08-26', '2016-09-28', '2016-09-29', 
     '2016-09-30', '2016-10-26', '2016-10-27', '2016-10-28', 
     '2016-11-16', '2016-11-17', '2016-11-18'], dtype='datetime64[D]') 

In [193]: (t[:,None] + [-1,0,1]).ravel() 
Out[193]: 
array(['2016-07-27', '2016-07-28', '2016-07-29', '2016-08-24', 
     '2016-08-25', '2016-08-26', '2016-09-28', '2016-09-29', 
     '2016-09-30', '2016-10-26', '2016-10-27', '2016-10-28', 
     '2016-11-16', '2016-11-17', '2016-11-18'], dtype='datetime64[D]') 
+0

非常好的解决方案。比我的好得多:-) – NinjaGaiden

+0

伟大的解决方案(一如既往!)。谢谢。 –

2

这个怎么样?

def f(d): 
    return (d-np.timedelta64(1,'D'),d,d+np.timedelta64(1,'D')) 

import numpy as np 
t = np.array(['2016-07-28', '2016-08-25', '2016-09-29', '2016-10-27', '2016-11-17'], dtype='datetime64[D]') 
print np.asarray(map(f,t)) 
+0

这也行得通,但我想,@Divakars的解决方案更有效率。谢谢 –

+0

毫无疑问。我甚至不知道你可以做到这一点。 – NinjaGaiden