python - pandas 查找日期属于哪半年

Question

我有以下带有 date 列的 pandas 数据框。如何添加指定日期所属半年的列？

Answer 1

将日期转换为日期时间，然后使用 numpy.where比较小于或等于:

df['date'] = pd.to_datetime(df['date'], dayfirst=True)

df['half year'] = np.where(df['date'].dt.month.le(6), 'H1', 'H2')
print (df)
        date half year
0 1993-09-09        H2
1 1993-09-11        H2
2 1994-01-23        H1
3 1993-03-18        H1

没有 numpy 的解决方案，更改掩码以获取更大的值，如 6，添加 1 并转换为字符串:

df['date'] = pd.to_datetime(df['date'], dayfirst=True)

df['half year'] = 'H' + df['date'].dt.month.gt(6).add(1).astype(str)
print (df)
        date half year
0 1993-09-09        H2
1 1993-09-11        H2
2 1994-01-23        H1
3 1993-03-18        H1