Как создать разреженный столбец с отформатированной датой

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

Answer 2

Вы можете проверить с помощью .loc, назначить с помощью Series

df.loc[df.index.day==1,'str_date']=pd.Series(df.index.strftime('%b-%Y'),index=df.index)
df
            str_date
obs_date            
2018-01-01  Jan-2018
2018-01-02       NaN
2018-01-03       NaN
2018-01-04       NaN
2018-01-05       NaN
2018-01-06       NaN
2018-02-01  Feb-2018