程序员最近都爱上了这个网站  程序员们快来瞅瞅吧！  it98k网:it98k.com

I want to plot the number of spherical wavefronts I input using Python, this is my code.I'm using Axes3D to plot, but I have a problem, when I plot with "ax.plot_surface(x, y, z, facecolors=plt.cm.coolwarm(w), shade=False)" doesn´t work and appears me the next message:

ValueError: Argument Z must be 2-dimensional.

My code is the following:

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

def spherical_wavefront(n):
    x, y, z = np.meshgrid(np.linspace(-1, 1, n), np.linspace(-1, 1, n), np.linspace(-1, 1, n))
    r = np.sqrt(x**2 + y**2 + z**2)
    wavefront = np.empty((n, n, n))
    wavefront[r <= 1] = 1/r[r <= 1] * np.exp(1j * r[r <= 1])
    return x, y, z, np.real(wavefront)

num_wavefronts = int(input("How many spherical wavefronts would you like to plot? "))

fig = plt.figure()

for i in range(num_wavefronts):
    x, y, z, w = spherical_wavefront(100)
    ax = fig.add_subplot(111, projection='3d')
    ax.plot_surface(x, y, z, facecolors=plt.cm.coolwarm(w), shade=False)
    ax.set_title(f'Spherical Wavefront {i + 1}')
    ax.set_xlabel('X')
    ax.set_ylabel('Y')
    ax.set_zlabel('Z')

plt.show()

解决方案

暂无回答