熊猫-产生最后一位数字后的下一个最近年份

在计算某些金融工具的到期日时,通常我们需要做的只是以下几点:


Trade Date  RIC

5/22/1989   SPH0

5/23/1989   SPH0

5/24/1989   SPH0

5/25/1989   SPH0

5/26/1989   SPH0

如果交易日期是工具的交易日,那么RIC是4个字母的字符串,其中包括:


First 2 characters = an asset class

3rd character = expiration month

last character = last digit of expiration year

到期月份说明:


month_codes_to_int = {'F':'1', 'G':'2', 'H':'3', 'J':'4', 'K':'5', 'M':'6', 

             'N':'7', 'Q':'8', 'U':'9', 'V':'10', 'X':'11', 'Z':'12'}

我通过使用交易日期和RIC为每一行生成一个到期年...我这样做是通过遍历DF并每次使用将正确计算到期日期的函数填充列中的空白单元格来实现的。


for index, row in df.iterrows():

    row['Trade Date'] = pd.to_datetime(row['Trade Date'])

    print(row['Trade Date'], row['RIC'])

    current_year = row['Trade Date'].year

    asset_class = row['RIC'].split[0:3]

    expiration_month = row['RIC'][2]

    expiration_year_last_digit = row['RIC'][3]

    expiration_year = 

我的方法是让到期日期是与当前日期相距最近的月份和年份,因此,以1989年5月22日为例,该日期应为1990年3月15日(日期无关紧要,1990年为最接近的日期)年份从0到1989年结束)。


有没有一种方法可以使用pandas datetime功能自动执行此操作?


临摹微笑
浏览 123回答 1
1回答

绝地无双

首先,创建包含月份,年份以及TradeDate列的最后一位的系列。m = df.RIC.str[2].map(month_codes_to_int)y = df.RIC.str[3].astype(int)s = df.TradeDate.dt.year.mod(10)然后计算您的偏移量:offset = np.where(y==s, 0, 10+y-s)最后,创建您的新列:pd.to_datetime((df.TradeDate.dt.year + offset).astype(str) + m, format='%Y%m')输出:0   1990-03-011   1990-03-012   1990-03-013   1990-03-014   1990-03-01dtype: datetime64[ns]
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python