python - 根据前一行 python 数据框中存在的年份格式化月份

Question

对于下面的内容，有没有办法根据索引 0 中的值将 12 月之前的月份格式化为“DEC-2017”，并将 12 月之后的月份格式化为“JAN-2018”？我使用的是 Python 2.7。

import pandas as pd
df = pd.DataFrame([['2017-18','','','','','','','','','','',''], ['APR', 'MAY', 'JUN', 'JULY', 'AUG', 'SEP', 'OCT', 'NOV', 'DEC', 'JAN', 'FEB', 'MAR']])

输出应如下所示:

df1 = pd.DataFrame([['2017-18','','','','','','','','','','',''], ['APR-2017', 'MAY-2017', 'JUN-2017', 'JULY-2017', 'AUG-2017', 'SEP-2017', 'OCT-2017', 'NOV-2017', 'DEC-2017', 'JAN-2018', 'FEB-2018', 'MAR-2018']])

抱歉，我是新人，所以不知道如何在此处粘贴数据框，从而使用代码。

Answer 1

cumsum + iloc

df.iloc[1] += '-' + (df.iloc[1].eq('JAN').cumsum() + 2017).astype(str)

df
         0         1         2          3         4         5         6   \
0   2017-18                                                                
1  APR-2017  MAY-2017  JUN-2017  JULY-2017  AUG-2017  SEP-2017  OCT-2017   

         7         8         9         10        11  
0                                                    
1  NOV-2017  DEC-2017  JAN-2018  FEB-2018  MAR-2018