python numpy.datetime64 数组：按月过滤

Question

假设我有这个数据集（或更广泛的一个）：

import numpy as np

array = np.array([
    np.datetime64('2022-01-03'),
    np.datetime64('2022-05-03'),
    np.datetime64('2022-12-03')
])

我想按月份将日期分成不同的 arrays。 我怎样才能做到这一点？

一个相关且更普遍的问题是如何按年，月，日过滤它们......

Answer 1

如果您希望它们在 DataFrame 中结构化，您可以这样做：

    array = np.array([
        np.datetime64('2022-01-03'),
        np.datetime64('2022-05-03'),
        np.datetime64('2022-12-03')
    ])

dates=pd.DataFrame()
dates["date"]=pd.to_datetime(array)
for i in range(1,13):
    dates[f"month_{i}"]=[x if x.month==i else np.nan for x in list(dates["date"])]    

Output：

        date    month_1  month_2  ...  month_10  month_11   month_12
0 2022-01-03 2022-01-03      NaN  ...       NaN       NaN        NaT
1 2022-05-03        NaT      NaN  ...       NaN       NaN        NaT
2 2022-12-03        NaT      NaN  ...       NaN       NaN 2022-12-03

[3 rows x 13 columns]