mpl_method_chaining.py

import matplotlib.pyplot as plt
from pydataset import data

class Plot:
    def __init__(self, data=None):
        if data is not None:
            self.data = data
    def __getattr__(self, attr):
        def wrapper(*args, **kwargs):
            if attr in ['scatter', 'hist', 'bar', 'plot'] and 'data' not in kwargs:
                kwargs['data'] = self.data
            getattr(plt, attr)(*args, **kwargs)
            return self
        return wrapper

tips = data('tips')

Plot(tips).scatter('total_bill', 'tip')