Question

我正在尝试通过以下程序使用matplotlib.pyplot.contourf()绘制3D图表：

import numpy as np
import matplotlib.pyplot as plt
import scipy

# calculates Fast Fourier transforms for each value in the 1D array "Altitude"
# and stacks them vertically to form a 2D array of fft values called "Fourier"
Fourier = np.array([])
for i in range(len(Altitude)):
    Ne_fft = Ne_lowpass[i,:]/np.average(Ne_lowpass[i,:])
    Ne_fft = Ne_fft - Ne_fft.mean()
    W = scipy.fftpack.fftfreq(10*Ne_fft.size, d=(Time[-1]-Time[0])/len(Ne_fft))
    P = 1/abs(W)
    FFT = abs(scipy.fftpack.fft(Ne_fft, n=10*len(Ne_fft)))
    FFT = FFT**2
    if len(Fourier) == 0:
        Fourier = FFT
    else:
        Fourier = np.vstack((Fourier,FFT))

# plots the 2D contourf plot of "Fourier", with respect to "Altitude" and period "P"
plt.figure(5)
C = plt.contourf(P,Altitude,Fourier,100,cmap='jet')
plt.xscale('log')
plt.xlim([1,P[np.argmax(P)+1]])
plt.ylim([59,687])
plt.ylabel("Altitude")
plt.xlabel("Period")
plt.title("Power spectrum of Ne")
cbar = plt.colorbar(C)
cbar.set_label("Power", fontsize = 16)

在大多数情况下，它运行良好；但是，在某些地方绘制了无用的空白区域。可以找到here生成的剧情（对不起，我没有足够的声誉点来直接附加图像）

该程序的目的是在二维numpy数组的1个轴上计算一系列快速傅立叶变换，并将它们堆叠起来以显示轮廓图，描绘出哪些周期性在数据中最突出。

我检查了标绘数量中显示为白色的部分，并且仍然存在有限值，尽管比图中其他地方的显着数量小得多。

print(Fourier[100:,14000:])
[[  2.41147887e-03   1.50783490e-02   4.82620482e-02 ...,   1.49769976e+03
    5.88859945e+02   1.31930217e+02]
 [  2.12684922e-03   1.44076962e-02   4.65881565e-02 ...,   1.54719976e+03
    6.14086374e+02   1.38727145e+02]
 [  1.84414615e-03   1.38162140e-02   4.51940720e-02 ...,   1.56478339e+03
    6.23619105e+02   1.41367042e+02]
 ..., 
 [  3.51539440e-03   3.20182148e-03   2.38117665e-03 ...,   2.43824864e+03
    1.18676851e+03   3.13067945e+02]
 [  3.51256439e-03   3.19924000e-03   2.37923875e-03 ...,   2.43805298e+03
    1.18667139e+03   3.13042038e+02]
 [  3.50985146e-03   3.19677302e-03   2.37741084e-03 ...,   2.43790243e+03
    1.18659640e+03   3.13021994e+02]]

print(np.isfinite(Fourier.all()))
True
print(np.isnan(Fourier.any()))
False

是否存在空白空间，因为值与绘图的其余部分相比是如此之小？我完全不确定如何解决此问题。

Answer 1

在plt.contourf(P,Altitude,Fourier,100,cmap='jet')行中，您正在获取100个自动选择的等高线图。在这种情况下，“自动”不能保证这些级别包括所有数据。

如果要确保包括所有数据，可以定义自己的级别使用

plt.contourf(x, y, Z, np.linspace(Z.min(), Z.max(), 100))

Answer 2

您可以通过添加选项extend ='both'来解决此问题。例如。 C = plt.contourf（P，Altitude，Fourier，100，cmap ='jet'，extend ='both'）

参考：https://matplotlib.org/examples/pylab_examples/contourf_demo.html

Contourf（）在有限数据上绘制空白

2 个答案: