преобразовать год-месяц-число в год-квартал в pandas

Question

У меня есть фрейм данных pandas, где один столбец (first_date) содержит дату в следующем формате: 2018-03-31

#I have created a monthly level data using the below code
    dataset['first_date']=pd.to_datetime(dataset['first_date']) 
    dataset['first_date'].apply(lambda x: x.strftime('%Y-%m'))
    **output: 2018-03**

Мой вопрос : с использованием first_date, которое я хочу чтобы создать новый столбец, чтобы получить год и квартал. выведите что-то вроде 2018-Q1 и должно быть в строке, а не в периодах

Answer 1

Попробуйте следующее:

df['first_date'] = pd.to_datetime(df['first_date']).dt.strftime('%Y-%m')
df['quarter'] = pd.to_datetime(df['first_date']).dt.to_period('Q').astype(str).str.replace('Q', '-Q')
print(df)

 first_date  quarter
0    2018-03  2018-Q1
1    2018-03  2018-Q1
2    2018-03  2018-Q1
3    2018-03  2018-Q1
4    2018-03  2018-Q1
5    2018-05  2018-Q2
6    2018-09  2018-Q3