Python 3d打印集固定颜色比例

Question

我试图绘制两个3d数组：第一个数组的（z）值在[-0.15, 0.15]范围内，第二个数组的值在[-0.001, 0.001]范围内。当我打印时，色标会自动跟随数据范围：如何设置自定义比例？我不想看到0.001的浅色，而应该看到0.15的浅色。我怎样才能修秤

我共享以下部分代码：

import numpy as np 
import matplotlib.pyplot as plt 
import plotly.graph_objects as go


data1=np.load("test.npy")
fig = go.Figure(data=[go.Surface(z=data1,colorscale="viridis")])


fig.update_layout(margin = dict(t=100,r=100,b=100,l=100),
    scene = dict(zaxis = dict(range=[-0.15,0.15],)),
    showlegend = False,
    width = 1000, height = 1000,
    autosize = False )
fig.update_layout(scene = dict(
                    xaxis_title='x',
                    yaxis_title='y',
                    zaxis_title='z'))
fig.write_image("image.pdf")
fig.show()

Answer 1

我刚刚查阅了plotly API引用，了解了Surface需要什么样的参数。在那里我看到{}、{}和{}。您可以设置这些值来告诉曲面颜色比例的最小值、最大值和中点应该在哪里。在您的情况下，只需将cmin和cmax设置为适合您的用例的适当值：

例如go.Surface(z=data1, colorscale="viridis", cmin=-0.15, cmax=0.15)