将3d色阶绘制成径向坐标而不是z

Question

默认情况下，绘制三维表面图将颜色标度映射到z坐标。现在我试图用图表绘制三维辐射图，如你所见[{3}}。在这种情况下，理想的是将颜色标度映射到径向坐标而不是z，因为在这种情况下z没有实际的物理意义，而径向坐标表示给定方向的辐射强度。因此问题是：是否可以将绘图颜色重新映射到新的numpy数组，在这种情况下它将代表径向坐标？下面报告了用于生成绘图的代码，其中m_x, m_y, m_z是包含数据的numpy数组。

data = go.Data([go.Surface(
                       x=m_x,
                       y=m_y,
                       z=m_z)])
layout = go.Layout(
title='radiation_pattern',
autosize=True,
width=800,
height=800,
margin=go.Margin(
    l=65,
    r=50,
    b=65,
    t=90
    )
)
fig = go.Figure(data=data, layout=layout)
iplot(fig)