Python三维绘图设置固定的色彩比例

我试图绘制两个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()

问题来源StackOverflow 地址：/questions/59379173/python-3d-plot-set-fixed-colorscales