colored wireframe plot in matplotlib

Question

I am trying to color a wireframe plot according to the z-value. I can\'t find any code examples on the internet.

Here is an example of a surface plot that has the co

Answer 1

I had a similar problem with coloring and sizing circles according to a variable which did not work either. So my workaround was to bin the values of the variables and loop over the bins. I masked the data such that the array mask did only contain the data with values in that bin.

ax.plot_wireframe(mask[i], ..., color="red")
ax.plot_wireframe(mask[i], ..., color="blue") 
etc.

I know it's not very elegant but in my case it did the job ;)