python Pandas : How do I apply a function requiring an extended class (datetime)?

Question

如何将 pandas apply 用于需要扩展标准类 (datetime) 的函数？

具体来说，我想从 https://gist.github.com/treyhunner/6218526 的优秀要点中导入 datetime_modulo .

此代码扩展了标准 datetime 类，以允许将模运算应用于 datetime 对象，例如

from datetime_modulo import datetime
from datetime import timedelta
d = datetime.now()
print d % timedelta(seconds=60)

现在我需要将此模运算应用到 pandas DataFrame 列/系列，例如

df['dates'] = pd.to_datetime(df.index.values)
df['datetime_mod'] = df['dates'].apply(lambda x: x % timedelta(minutes=15))

但是 pandas 无法检测到扩展的日期时间类(除非我只是错误地使用它):

TypeError: unsupported operand type(s) for %: 'Timestamp' and 'datetime.timedelta'

如何进行？

Answer 1

你可以试试，按照this suggestion ，将操作数显式转换为 datetime:

from datetime_modulo import datetime
from datetime import timedelta

df = pd.DataFrame({'Time': [pd.to_datetime('now')]})

def modulo(x):
    dt = datetime(year=x.year,month=x.month,day=x.day, hour=x.hour, minute=x.minute, second=x.second)
    return dt % timedelta(seconds=60)

df['Time'] = df['Time'].apply(modulo)