如何添加装饰器,声明函数的传入pandas dataframe参数具有日期时间索引？
我看过包engarde和validada,但还没找到任何东西.我可以在函数内部进行检查,但更喜欢装饰器.
解决方法:
正如@PadraicCunningham所写,使用functools.wraps创建一个并不太难：
import functools
def assert_index_datetime(f):
@functools.wraps(f)
def wrapper(df):
assert df.index.dtype == pd.to_datetime(['2013']).dtype
return f(df)
return wrapper
@assert_index_datetime
def fn(df):
pass
df = pd.DataFrame({'a': [1]}, index=pd.to_datetime(['2013']))
fn(df)
版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 [email protected] 举报，一经查实，本站将立刻删除。