Color options on 3d Surface Plot

I am working with many-dimensional data and and was drawn to Plotly because of the great color visualization. After some experimentation I wasn’t able to figure it out on my own - is there a way to base the color off a different variable than one of the other axes?

Ideally, I would have a heatmap-style surface plot where x and y are two different categorical variables, z is the number of counts, and color is a third categorical variable.

Thanks!

Try surfacecolor http://codepen.io/etpinard/pen/yMOmoy