Plotly堆叠的表面共享相同的颜色条

时间:2018-01-09 00:15:51

标签: python plot plotly

我想绘制共享相同颜色条和色阶的3个堆叠表面。 我希望彩条在[0.0,0.1,...,0.9,1.0]中有11个刻度。

出于某种原因,我似乎无法在三个图表中使用相同的颜色条和比例。现在他们都有自己的色阶。 我可以让它只打印一次彩色条,但它们仍然会有不同的比例。

这是我的代码:

def plot_3d(plot_dict, title):
data = []
for i in range(3):
    dataset = plot_dict[kernels[i]]
    data.append(
        go.Surface(
            x=C_2d_range,
            y=gamma_2d_range,
            z=dataset, 
            #showscale=True if i == 0 else False, 
            colorbar=dict(
                nticks=11, 
                tickmode='array',
                tickvals=[0.0, 0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1.0]
            ),
            colorscale='Viridis'
        )
    )

layout = go.Layout(
    title=title,
    scene = dict(
        xaxis = dict(
            title='C parameter'
        ),

        yaxis = dict(
            title='Gamma parameter'
        ),

        zaxis = dict(
            title='F1 Score'
        )
    )
)

fig = go.Figure(data=data, layout=layout)

ply.iplot(fig, filename=title)  

但它会产生this,其中每个情节都清楚地使用自己的比例。为了清楚这一点,我想在底部表面是黑色的,而顶部的表面是黄色的。

感谢您的帮助。

1 个答案:

答案 0 :(得分:1)

这个问题还很晚,但对于将来的读者:

要实现此目的,您需要为所有表面对象设置相同的cmincmax

import numpy as np

import plotly.plotly as py
import plotly.graph_objs as go
from plotly.offline import plot

z1 = np.random.uniform(0, 1, (10, 10))
z2 = z1 + 1
z3 = z1 - 1
min_value = np.r_[z1,z2,z3].min()
max_value = np.r_[z1,z2,z3].max()

surf1 = go.Surface(z=z1, cmin=min_value, cmax=max_value)
surf2 = go.Surface(z=z2, showscale=False, cmin=min_value, cmax=max_value)
surf3 = go.Surface(z=z3, showscale=False, cmin=min_value, cmax=max_value)

data = [surf1, surf2, surf3]

plot(data,filename='multiple-surfaces')

结果: Sample plot