Passing extra arguments to heatmapf callable

[vegardjervell]

First of all i want to say this all looks great! I needed to pass some extra arguments to the function i was making a heatmap for, so i added it as a kwarg to the heatmapf function (same style as scipy.integrate.quad() and other functions that take in callables).

I think it would be a nice little addition to the library. Linking a pull request where I've implemented it.

[SpyrosBl]

Hello, I saw your post, implemented your changes to the base code and wanted some help or a clarification. I am trying to implement a function, similar to the "Shannon Entropy" example given in the basic documentation page. The only difference is that I pass two variables p, q to the function instead of only p (as in the Shannon). I do come across to the following error " missing 1 required positional argument: 'q' " . But if I leave out q, and return to the single value function everything seems to work. Any ideas?

[vegardjervell]

Hi, awesome! I've written a little example here of how I've been using it locally in my project,

def some_func(x, args):
    a, b, c = args
    x, y, z = x
    return x*a + y*b + z*c

scale=60
fig, tax = tr.figure(scale=scale)
tax.heatmapf(some_func, scale=scale, boundary=False, fargs=(1,-2,3), vmin=-2, vmax=3)
plt.show()

So i just pass any additional arguments as a tuple, and pack them out inside the function. I'm a bit new to Github btw, so I didn't immediately see how i could link my pull request to this issue.

[SpyrosBl]

You are a Top Guy, many thanks, now I understand your implementation!

[vegardjervell]

Thanks! Feel free to add the example to the docs, if you didn't understand it right away, most other people probably wont either :)

[SpyrosBl]

Will certainly do!