Plotの描画関連

複数の時系列を縦に並べてplotする

ここに書いてある、gwpy.plotのPlot()がオススメ。
GUIで拡大等をしたとき、他のaxesも一緒に動く。
注意点は、TimeSeriesのlistやTimeSeriesDictを引数にしてまとめて渡した場合は separate=True をしても分割されないという謎仕様。面倒だが1つ1つ手で並べる必要がある。

   1 from gwpy.plot import Plot
   2 plot = Plot(l1hoft, h1hoft, separate=True, sharex=True)
   3 plot.show()

戻り値 (ここでは plot)は、matplotlibのfigureだと思ってしまってよい。
なので、それぞれのy軸ラベルや全体のタイトルを付けるには以下のようにする。

   1 import matplotlib.pyplot as plt
   2 axes = plot.get_axes()
   3 ylabel = ['y1','y2']
   4 for i, ax in enumerate(axes): ax.set_ylabel(ylabel[i])
   5 plt.tight_layout()
   6 plt.subplots_adjust(top=0.95)
   7 plot.suptitle('title')

時刻の基準を手で設定する

ここを参照

   1 ax = plot.gca()
   2 ax.set_epoch(1126259462) # GPS timeを入れる

DQ Flagを添える

.add_segments_bar() を使う(ここを参照)。
他にも .add_state_segments() や .add_dataqualityflag() といった似たものがある。詳細はPlotting APIを参照。

スペクトルのパーセンタイル/σバンドを付ける

ここを参照。手順としては、

まずTimeSeriesからspectrogram2()を使ってnon-averaged power Spectrogramを作る。
.percentile() でパーセンタイルのスペクトルを3つ作る。
plot_mmm() で描画するとバンド付きでplotされる。

   1 from gwpy.timeseries import TimeSeries
   2 from gwpy.plot import Plot
   3 
   4 hoft = TimeSeries. ...
   5 sg = hoft.spectrogram2(fftlength=4, overlap=2, window='hanning') ** (1/2.)
   6 median = sg.percentile(50)
   7 low = sg.percentile(5)
   8 high = sg.percentile(95)
   9 
  10 plot = Plot()
  11 ax = plot.gca(xscale='log', xlim=(10, 1500), xlabel='Frequency [Hz]',  yscale='log', ylim=(3e-24, 2e-20),  ylabel=r'Strain noise [1/$\sqrt{\mathrm{Hz}}$]')
  12 ax.plot_mmm(median, low, high, color='gwpy:ligo-hanford')
  13 ax.set_title('LIGO-Hanford strain noise variation around GW170817', fontsize=16)
  14 plot.show()

パーセンタイルではなくmeanとσで表したい場合は、Spectrogramの2次元配列に対して時間軸(0軸)方向にmean, stdを取ればよい。

   1 mean  = FrequencySeries(data=sg.mean(0), frequencies=sg.frequencies)
   2 sigma = FrequencySeries(data=sg.std(0),  frequencies=sg.frequencies)

matplotlib色々

プロットの複雑なレイアウトはGridSpecが便利かも、という話

matplotlibで一定区間に背景色をつける方法