def f(x, y): return (x**2 + y**2 - 1) * (x**2 + (x**3/3 - y**2/2)**2 - 1) * (y**2 + (x**3/3 - y**2/2)**2 - 1) - 1 x = np.linspace(-3, 3, 100) y = np.linspace(-3, 3, 100) X, Y = np.meshgrid(x, y) Z = f(X, Y) fig = go.Figure(data=[go.Surface(x=X, y=Y, z=Z)]) fig.show()这个图我想要画出来z轴只在-5到5,如何修改
时间: 2023-10-21 11:31:18 浏览: 128
A*算法的Python实现-堆优化版
你可以在创建 Surface 对象时,将 zmin 和 zmax 参数设置为 -5 和 5,以限制 z 轴的范围。修改后的代码如下:
```python
import numpy as np
import plotly.graph_objects as go
def f(x, y):
return (x**2 + y**2 - 1) * (x**2 + (x**3/3 - y**2/2)**2 - 1) * (y**2 + (x**3/3 - y**2/2)**2 - 1) - 1
x = np.linspace(-3, 3, 100)
y = np.linspace(-3, 3, 100)
X, Y = np.meshgrid(x, y)
Z = f(X, Y)
fig = go.Figure(data=[go.Surface(x=X, y=Y, z=Z,
zmin=-5, zmax=5)])
fig.show()
```
运行后就可以得到 z 轴范围在 -5 到 5 的图像了。
阅读全文