Как добавить плоскость y = x на трехмерный график поверхности в Plotly? ⇐ Python

[Anonymous]

В настоящее время я работаю над трехмерным графиком поверхности, используя Plotly в Python. Ниже приведен код, который у меня есть на данный момент:

Код: Выделить всё

import numpy as np
import plotly.graph_objects as go

# Definition of the domain
x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
X, Y = np.meshgrid(x, y)

# Definition of the function, avoiding division by zero
Z = np.where(X**2 + Y**2 != 0, (X * Y) / (X**2 + Y**2), 0)

# Creation of the interactive graph
fig = go.Figure(data=[go.Surface(z=Z, x=X, y=Y, colorscale='Viridis')])

# Add title and axis configurations
fig.update_layout(
title='Interactive graph of f(x, y) = xy / (x^2 + y^2)',
scene=dict(
xaxis_title='X',
yaxis_title='Y',
zaxis_title='f(X, Y)'
),
)

# Show the graph
fig.show()

Я хотел бы добавить к этому графику плоскость (y = x). Однако мне трудно понять, как это сделать.
Может ли кто-нибудь дать рекомендации о том, как добавить эту плоскость к существующему графику поверхности? Любая помощь будет принята с благодарностью!

Подробнее здесь: https://stackoverflow.com/questions/791 ... -in-plotly