在 Pandas DataFrame 上扩展绘图

cod*_*000 5 python pandas

我想扩展绘图函数以将自定义绘图函数与现有的 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)

谢谢