python画三维函数图_三维Python谱图（类似matlab的谱图函数）

我的问题是：

我有谱图（scipy.fftpack.fft）所需的所有值。我想用python创建一个3D谱图。在

在MATLAB中，这是一个非常简单的任务，而在python中则显得复杂得多。我试过mayavi，3D打印matplotlib，但我没能做到。在

谢谢

我的代码：import numpy as np

import pandas as pd

import numpy as np

from scipy import signal

import matplotlib.pyplot as plt

from mpl_toolkits.mplot3d import Axes3D

from matplotlib import cm

from matplotlib.collections import PolyCollection

fs = 11240.

t = 10

time = np.arange(fs*t) / fs

frequency = 1000.

mysignal = np.sin(2.0 * np.pi * frequency * time)

nperseg = 2**14

noverlap = 2**13

f, t, Sxx = signal.spectrogram(mysignal, fs, nperseg=nperseg,noverlap=noverlap)

myfilter = (f>800) & (f<1200)

fig,ax = plt.subplots()

plt.pcolormesh(t, f[myfilter], 10*np.log10(Sxx[myfilter, :]), cmap='jet')

plt.show()

fig = plt.figure()

ax = fig.gca(projection='3d')

x = []

y = []

for counter,i in enumerate(f):

x.append(np.array([i for k in t]))

y.append(t)

ax.plot_surface(np.array(x), np.array(y), 10.0*np.log10(Sxx), cmap=cm.coolwarm)

plt.show()