Как построить трехмерную поверхность?

Question

Я хочу построить трехмерную поверхность, которая имеет форму параметра:

x = (-2 * (s^2 - t)^  3)

y = log(t*(t-2*s^2)+2*s^2)

z = 1/(s^2-t)

, где s, t - параметр. и я использую plt_surface, но это не работает. Я не знаю, что происходит, не могли бы вы помочь мне?

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

s,t=symbols('s,t')
x=Function('x')(s,t)
z=Function('z')(s,t)
y=Function('y')(s,t)
figure = plt.figure()
ax = figure.add_subplot(1, 1, 1, projection='3d')
t1= np.linspace(1,100,200)
s1 = np.linspace(1,100,200)
s,t=np.meshgrid(s1,t1)
x = (-2 * (s*s - t)**3)
y = log(t*(t-2*s*s)+2*s*s)
z = 1/(s*s-t)
ax.plot_surface(x,y,z,cmap='rainbow')
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('z')
ax.set_title('f')
plt.show()

после запуска, это показывает:

Traceback (most recent call last):
  File "C:\Users\apple\Desktop\pywork\venv\lib\site-packages\sympy\core\cache.py", line 94, in wrapper
    retval = cfunc(*args, **kwargs)
TypeError: unhashable type: 'numpy.ndarray'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "C:\Users\apple\Desktop\pywork\venv\lib\site-packages\sympy\core\cache.py", line 94, in wrapper
    retval = cfunc(*args, **kwargs)
TypeError: unhashable type: 'numpy.ndarray'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):


     File "C:/Users/apple/Desktop/pywork/mining/rep.py", line 19, in <module>
        y = log(s)
      File "C:\Users\apple\Desktop\pywork\venv\lib\site-packages\sympy\core\cache.py", line 96, in wrapper
        retval = func(*args, **kwargs)
      File "C:\Users\apple\Desktop\pywork\venv\lib\site-packages\sympy\core\function.py", line 458, in __new__
        result = super(Function, cls).__new__(cls, *args, **options)
      File "C:\Users\apple\Desktop\pywork\venv\lib\site-packages\sympy\core\cache.py", line 96, in wrapper
        retval = func(*args, **kwargs)
      File "C:\Users\apple\Desktop\pywork\venv\lib\site-packages\sympy\core\function.py", line 277, in __new__
        evaluated = cls.eval(*args)
      File "C:\Users\apple\Desktop\pywork\venv\lib\site-packages\sympy\functions\elementary\exponential.py", line 577, in eval
        coeff = arg.as_coefficient(S.ImaginaryUnit)
    AttributeError: 'ImmutableDenseNDimArray' object has no attribute 'as_coefficient'

    Process finished with exit code 1

Answer 1

Чтобы получить какой-то результат из вашего кода, вам нужны правильные входные данные. В этом случае домен t1 не подходит и вызывает ошибки в вычислениях y и z. Вот модифицированный код, в котором домен t1 изменяется таким образом, что исходный код может работать и создавать некоторый график.

from matplotlib import pyplot as plt
import numpy as np
#import math
import mpl_toolkits.mplot3d.axes3d as axes3d
from sympy import symbols, Function

s,t = symbols('s,t')
x = Function('x')(s,t)
z = Function('z')(s,t)
y = Function('y')(s,t)

figure = plt.figure(figsize=(8,8))
ax = figure.add_subplot(1, 1, 1, projection='3d')

# experiment with various range of data
t1 = np.linspace(-1,-100,200)  #...to enable computation
s1 = np.linspace(1,100,200)
s,t = np.meshgrid(s1,t1)
x = -2*(s*s - t)**3

y = np.log(t*(t-2*s*s)+2*s*s)  # need np.
z = 1/(s*s-t)  

ax.plot_surface(x,y,z,cmap='rainbow')
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('z')
ax.set_title('f')
plt.show()

Выходной график: