画两个数的平方和的曲线

代码1：

from mpl_toolkits import mplot3d
import numpy as np
import matplotlib.pyplot as plt# Creating dataset
x = np.outer(np.linspace(-3, 3, 32), np.ones(32))
y = x.copy().T # transpose
z = (x **2 + y **2 )# Creating figure
fig = plt.figure(figsize =(14, 9))
ax = plt.axes(projection ='3d')# Creating plot
ax.plot_surface(x, y, z)# show plot
plt.show()

效果：

代码2：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D# Define the mathematical function to plot
def f(x, y):return (x**2 + y**2)# Generate x and y values
x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
X, Y = np.meshgrid(x, y)
Z = f(X, Y)# Create a 3D plot
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')# Plot the surface
surf = ax.plot_surface(X, Y, Z, cmap='viridis')# Customize the plot
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
ax.set_title('3D Surface Plot of x^2 + y^2')# Add a color bar
fig.colorbar(surf)# Show the plot
plt.show()

 效果：