3D-плоттер работает только для определенных форм

Question

Проблема в том, что этот сценарий не может построить сферу, например, в то время как он может построить несколько конусов, например, один в сценарии.

Я изменил форму и попытался найти линиииз которого возникает ошибка при использовании сообщения об ошибке, полученного при построении сферы.

import sympy as sy
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits import mplot3d

# Plot Figure
nd = 50  # Number of points in graph

ax = plt.axes(projection='3d')  # Adds 3d axis to figure

x1 = np.linspace(-15, 15, nd)
y1 = np.linspace(-15, 15, nd)

X, Y = np.meshgrid(x1, y1)  # Create 2D grid with x1 and y1

i = 0
a = 0
b = 0
Z = np.array([])

x = sy.Symbol('x')
y = sy.Symbol('y')
z = (x**2+y**2)**0.5  # Function goes here

for i in range(nd):  # Iterate over rows
    b = 0
    xv1 = X[a, :]
    yv1 = Y[a, :]
    for i in range(nd):  # Iterate over elements in one row
        xv = xv1[b]
        yv = yv1[b]
        z1 = z.subs([(x, xv), (y, yv)])
        Z = np.append(Z, z1)  # Append values to array just a row
        b = b + 1
    a = a + 1

Z = np.reshape(Z, (nd, nd))

print(Z.dtype)
print(Y.dtype)
print(X.dtype)
Z = Z.astype(float)
Y = Y.astype(float)
X = X.astype(float)

ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap='viridis', edgecolor='none')
plt.show()

В результате выполнения функции сферы происходит сбой сценария.Я надеюсь, что этот скрипт сможет отображать такие 3D-формы.

Answer 1

Это ошибка, которую вы получаете случайно?

TypeError: can't convert complex to float

Как это сформулировано, вы запрашиваете мнимое число обратно.Если вы определите это как свое сферное уравнение:

z = (x**2+y**2-1)**0.5

, вы в конечном итоге попросите sqrt (-1), когда x = y = 0, что не будет работать.Попробуйте параметризовать со сферическими координатами, как в этом примере: Python / matplotlib: построение трехмерного куба, сферы и вектора?