python - 从 numpy datetime64 获取年、月或日

Question

我有一个 datetime64 类型的数组:

dates = np.datetime64(['2010-10-17', '2011-05-13', "2012-01-15"])

有没有比遍历每个元素更好的方法来获取 np.array 的年份:

years = f(dates)
#output:
array([2010, 2011, 2012], dtype=int8) #or dtype = string

我使用的是稳定的 numpy 版本 1.6.2。

Answer 1

我发现与所描述的 pandas 方法相比，以下技巧可以将速度提高 2 倍到 4 倍 in this answer (即 pd.DatetimeIndex(dates).year 等)。我发现 [dt.year for dt in dates.astype(object)] 的速度与 pandas 方法相似。这些技巧也可以直接应用于任何形状(2D、3D 等)的 ndarray。

dates = np.arange(np.datetime64('2000-01-01'), np.datetime64('2010-01-01'))
years = dates.astype('datetime64[Y]').astype(int) + 1970
months = dates.astype('datetime64[M]').astype(int) % 12 + 1
days = dates - dates.astype('datetime64[M]') + 1