在python中绘制三个变量的函数

Question

我想绘制此功能的轮廓线，但我找不到任何有用的方法。

潜在的功能是： V(x,y,z) = cos(10x) + cos(10y) + cos(10z) + 2*(x^2 + y^2 + z^2)

我没有成功尝试类似的事情：

import numpy
import matplotlib.pyplot.contour

def V(x,y,z):
    return numpy.cos(10*x) + numpy.cos(10*y) + numpy.cos(10*z) + 2*(x**2 + y**2 + z**2)

X, Y, Z = numpy.mgrid[-1:1:100j, -1:1:100j, -1:1:100j]

但是，我不知道下一步如何绘制它？

matplotlib.pyplot.contour(X,Y,Z,V)

Answer 1

当您尝试传递contour三维数组时会出现错误，因为它需要二维数组。

考虑到这一点，请尝试：

from mpl_toolkits.mplot3d import axes3d
import matplotlib.pyplot as plt
from matplotlib import cm
import numpy as np

def V(x,y,z):
     return np.cos(10*x) + np.cos(10*y) + np.cos(10*z) + 2*(x**2 + y**2 + z**2)

X,Y = np.mgrid[-1:1:100j, -1:1:100j]
Z_vals = [ -0.5, 0, 0.9 ]
num_subplots = len( Z_vals)

fig = plt.figure( figsize=(10,4 ) )
for i,z in enumerate( Z_vals) :
    ax = fig.add_subplot(1 , num_subplots , i+1, projection='3d', axisbg='gray')
    ax.contour(X, Y, V(X,Y,z) ,cmap=cm.gnuplot)
    ax.set_title('z = %.2f'%z,fontsize=30)
fig.savefig('contours.png', facecolor='grey', edgecolor='none')

您可以使用contourf来显示表面，这在我看来看起来更好。

稍微详细说明（这可能超出了堆栈溢出的范围，但可能有助于解释前面的代码）：

正如您所知，没有直接的方法可视化3个变量的函数，因为它是一个生活在4个维度的对象（表面）。相反，必须使用函数切片来查看最新情况。通过切片，我的意思是将函数投影到较低维度空间。通过将一个或多个函数变量设置为常数来实现切片。