在Pandas DataFrame上扩展Plot

时间:2017-11-16 10:50:02

标签: 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)

谢谢

0 个答案:

没有答案