如何在Python中将xaxis网格放在频谱图上？

Question

我有下面的图,它提供压力信号的频谱图以及放在其上的信号用于比较.我能够在频谱图上绘制y轴网格,但无法在其上放置x轴网格.

用于生成频谱图的数据可在此处获得.

可重现的代码

from __future__ import division
from matplotlib import ticker as mtick
from matplotlib.backends.backend_pdf import PdfPages
import matplotlib.pyplot as plt
import numpy as np

data = np.genfromtxt('pressure.dat', skiprows = 1, delimiter = '\t')
pressure = data[:, 1]
theta = data[:, 0]


with PdfPages('Spectorgram of cylinder pressure.pdf') as spectorgram_pressure:
    _spectorgram_pressure_vs_frequency_ = plt.figure(figsize=(5.15, 5.15))
    _spectorgram_pressure_vs_frequency_.clf()
    spectorgram_pressure_vs_frequency = plt.subplot(111)
    cax = plt.specgram(pressure * 100000, NFFT = 256, Fs = 90000, cmap=plt.cm.gist_heat, zorder = 1)
    spectorgram_pressure_vs_frequency.grid(False, which="major")
    spectorgram_pressure_vs_frequency.set_xlabel('Time (s)', labelpad=6)
    spectorgram_pressure_vs_frequency.set_ylabel('Frequency (Hz)', labelpad=6)
    y_min, y_max = spectorgram_pressure_vs_frequency.get_ylim()
    # plt.gca
    cbar = plt.colorbar(orientation='vertical', ax = spectorgram_pressure_vs_frequency, fraction = 0.046, pad = 0.2)
    cbar.set_label('Power spectral density (dB)', rotation=90)
    primary_ticks = len(spectorgram_pressure_vs_frequency.yaxis.get_major_ticks())
    pressure_vs_time = spectorgram_pressure_vs_frequency.twinx()
    pressure_vs_time.plot(((theta + 360) / (6 * 6600)), pressure, linewidth = 0.75, linestyle = '-', color = '#FFFFFF', zorder = 2)
    pressure_vs_time.grid(b = True, which='major', color='#FFFFFF', linestyle=':', linewidth = 0.3)
    spectorgram_pressure_vs_frequency.xaxis.grid(True, which='major', color='#FFFFFF', linestyle=':', linewidth = 0.3)
    pressure_vs_time.set_ylabel('Cylinder pressure (bar)', labelpad=6)
    pressure_vs_time.yaxis.set_major_locator(mtick.LinearLocator(primary_ticks))
    spectorgram_pressure_vs_frequency.set_xlim([0, max(cax[2])])
    spectorgram_pressure.savefig(bbox_inches='tight')
    plt.close()

如何将x轴网格放在频谱图之上,就像Python中的y轴网格一样？我使用的是matplotlib 1.3.1版.这是特定于版本的问题吗？

更新

我将matplotlib从版本1.3.1更新到1.4.3,即使这样我也无法设置x轴网格.

Answer 1

正如其他人所指出的那样 - 使用您提供的代码复制您的问题非常困难.

特别是 - 我尝试过Windows 8.1,Ubuntu 14.04(在Virtualbox VM上),matplotlib版本1.3.1和1.4.3,有和没有text.usetexset以及Python 2.7.6和Python 3.它们都没有重现你的问题您提供的代码.

然而

如果我更换线路,我可以重现你所看到的

spectorgram_pressure_vs_frequency.xaxis.grid(True, which='major', color='#FFFFFF', linestyle=':', linewidth = 0.3)

同

pressure_vs_time.xaxis.grid(True, which='major', color='#FFFFFF', linestyle=':', linewidth = 0.3)

即我试图设置xaxis.grid孪生轴而不是原始轴.我必须得出结论,在你的真实代码中,你设置的xaxis.grid是你的孪生轴而不是你的主轴.

这意味着我可以回答您的直接问题如下:

如何将x轴网格放在频谱图之上,就像Python中的y轴网格一样？

您的代码执行此操作 - 您调用xaxis.grid()原始(非双联)轴.

我使用的是matplotlib 1.3.1版.这是特定于版本的问题吗？

不,1.3.1和1.4.3中的行为是相同的

Answer 2

虽然在我的情况下（python3.4，matplotlib 1.4.2，Ubuntu 14）我看到了x网格线，我想你可以尝试以下操作：

[pressure_vs_time.xaxis.get_gridlines()[x].zorder for x in range(len(pressure_vs_time.xaxis.get_gridlines())) ]

这将为您提供网格线的图层（上部具有更高的值，也可以是负值）。在我的例子中，所有行都返回 2。您可以设置：

[pressure_vs_time.xaxis.get_gridlines()[x].zorder = 1000 for x in range(len(pressure_vs_time.xaxis.get_gridlines())) ]