I'm trying to use matplotlib to visualise the continuous output of a function which takes 3 integers as arguments. I've iterated through each point in a 3d grid of inputs (like grid search in scikit-learn) with 1, 2 and 3 on each axis, calculating the function output for each combination, i.e. I calculate f(1,1,1)
, f(1,1,2)
, f(1,1,3)
, f(1,2,1)
and so on.
My question is, what is the best way to plot this output? I thought that, given the small range of each input (the range is 1-7, not 1-3, in my actual data), I could plot an ordinal series of heatmaps as planes in 3d, like so:
Is this possible with matplotlib?
Suggestions re. nicer ways to do the visualisation would also be welcome. This answer plots a 'heatcube' but, as the answerer points out, it's tricky to read.