python - 如何创建具有格式化日期的稀疏列

Question

我有一个时间序列数据集，索引为日期时间(一年中的每一天几乎有 1 行)，我希望每隔一段时间用一个字符串创建一个稀疏序列(用于绘图)。

obs_date    str_date
2018-01-01  Jan 2018
2018-01-02
2018-01-03
2018-01-04
2018-01-05
2018-01-06
...
2018-02-01 Feb 2018
etc

我试过了

df['str_date'] = df.index.strftime('%b\n%Y')
df.loc[df.index.day != 1, 'str_date'] = ''

并且可以正确设置月份的第一天，但​​是反过来试试

df['str_date'] = ''
df.loc[df.index.day == 1, 'str_date'] = df.index.strftime('%b %Y')

报错:ValueError:无法将输入数组从形状 (350) 广播到形状 (11)。有办法做到这一点吗？

Answer 1

问题是您作业的左右两侧大小不同。相反，您可以使用 pd.Index.where同时分配和屏蔽:

df['str_date'] = df.index.where(df.index.day == 1)

print(df) 
#              str_date
# obs_date             
# 2018-01-01 2018-01-01
# 2018-01-02        NaT
# 2018-01-03        NaT
# 2018-01-04        NaT
# 2018-01-05        NaT
# 2018-01-06        NaT
# 2018-02-01 2018-02-01

如果需要特定的字符串格式，可以调用strftime关于结果:

df['str_date'] = df.index.where(df.index.day == 1).strftime('%b-%Y')

print(df)
#             str_date
# obs_date            
# 2018-01-01  Jan-2018
# 2018-01-02       NaT
# 2018-01-03       NaT
# 2018-01-04       NaT
# 2018-01-05       NaT
# 2018-01-06       NaT
# 2018-02-01  Feb-2018