编写一段python 代码，画出函数: z=x3+ y3+x2+ y2+xy的 3D图

Question

#python#3D图
请编写一段python 代码，画出函数: z=x3+ y3+x2+ y2+xy的 3D图
(要求:x轴和y轴坐标刻度范围是(-20，20)，并且x轴和y轴的标签分别为:“x”“y”)

Answer 1

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

def func(x, y):
    return x**3 + y**3 + x**2 + y**2 + x*y

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

x = y = np.linspace(-1, 1, 100)
X, Y = np.meshgrid(x, y)
Z = func(X, Y)

ax.plot_surface(X, Y, Z, cmap='viridis')
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')

plt.show()