我想扩展绘图功能,将自定义绘图功能与现有的pandas绘图功能结合使用。这样做的最佳方式是什么?
如果我覆盖plot()
,我失去了正确使用旧功能的能力?或者有什么方法我仍然可以调用旧的绘图函数?
更新
我可以使用super()来调用父类,它看起来效果很好。如果这是一个糟糕的解决方案,请发表答案!
import pandas as pd
class CustomDataFrame(pd.DataFrame):
def plot(self, *args, **kwargs):
if 'kind' in kwargs and \
kwargs['kind'].lower() == 'dotmapplot':
return _custom_plot(*args, **kwargs)
else:
super(CustomDataFrame, self).plot(*args, **kwargs)
谢谢