我想扩展绘图函数以将自定义绘图函数与现有的 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)
Run Code Online (Sandbox Code Playgroud)
谢谢