相关疑难解决方法(0)

我编写了一个将pandas datetime日期转换为month-end的函数:

import pandas
import numpy
import datetime
from pandas.tseries.offsets import Day, MonthEnd

def get_month_end(d):
    month_end = d - Day() + MonthEnd() 
    if month_end.month == d.month:
        return month_end # 31/March + MonthEnd() returns 30/April
    else:
        print "Something went wrong while converting dates to EOM: " + d + " was converted to " + month_end
        raise

这个功能似乎很慢,我想知道是否有更快的替代方案？我注意到它很慢的原因是我在一个具有50'000个日期的数据帧列上运行它,并且我可以看到自引入该函数以来代码慢得多(在我将日期转换为月末之前).

df = pandas.read_csv(inpath, na_values = nas, converters = {open_date: read_as_date})
df[open_date] = df[open_date].apply(get_month_end)

我不确定这是否相关,但我正在阅读如下日期:

def read_as_date(x):
    return datetime.datetime.strptime(x, fmt)