python - numpy.arange 的日期时间间隔

Question

我想创建每月间隔而不使用 np.arange 进行迭代。作为一个简单的示例，我想要一个包含 np.datetime64 对象的数组，表示 1990-2000 范围内的元旦。我想是这样的:

np.arange(np.datetime64("1990-01-01"), np.datetime64("2000-01-01"), np.timedelta64(1, 'M'), dtype='datetime64[D]')

但这会引发错误

TypeError: Cannot cast NumPy timedelta64 scalar from metadata [M] to [D] according to the rule 'same_kind'

如果我将 dtype 切换为 datetime[M]，我会得到一个月份数组，但我当然想要日期。

Answer 1

您可以将原始dtype设置为datetime64[Y]，然后使用.astype将其转换为datetime64[D] ()

np.arange(np.datetime64("1990-01-01"), np.datetime64("2000-01-01"), np.timedelta64(1, 'Y'), dtype='datetime64[Y]').astype('datetime64[D]')