arbitrary_plot.py

from magic import integrate
functions_dict = {integrate:[(args),{kwargs}} # ask people about more elegant way to do this

def __init__(self, functions_dict):
    self.functions_dict = functions_dict

def run_analysis(self):
    for data in self.data:
        for func, v in self.functions_dict.items():
            ret = func(img, *v[0], *v[1])
            if isinstance(ret, tuple): # Then it is a 1D plot
                plot(ret[0], ret[1]
            else: # It is a 2D Plot
                imshow(ret)