python数据分析基础class 2(引力波的绘制)

 

 

 

引力波数据https://www.python123.io/dv/grawave.html

 

# 引力波

import numpy as np
import matplotlib.pyplot as plt
from scipy.io import wavfile

rate_h, hstrain = wavfile.read(r'H1_Strain.wav', 'rb')
rate_L, lstrain = wavfile.read(r'L1_Strain.wav', 'rb')
reftime, ref_H1 = np.genfromtxt('wf_template.txt').transpose()

htime_interval = 1 / rate_h
ltime_interval = 1 / rate_L

htime_len = hstrain.shape[0] / rate_h
htime = np.arange(-htime_len / 2, htime_len / 2, htime_interval)
ltime_len = lstrain.shape[0] / rate_L
ltime = np.arange(-ltime_len / 2, ltime_len / 2, ltime_interval)

fig = plt.figure(figsize=(12, 6))

plth = fig.add_subplot(2, 2, 1)
plth.plot(htime, hstrain, 'y')
plth.set_xlabel('Time(second)')
plth.set_ylabel('H1 Strain')
plth.set_title('H1_Strain')

pltl = fig.add_subplot(2, 2, 2)
pltl.plot(ltime, lstrain, 'g')
pltl.set_xlabel('Time(second)')
pltl.set_ylabel('L1 Strain')
pltl.set_title('L1_Strain')

pltref = fig.add_subplot(2, 1, 2)
pltref.plot(reftime, ref_H1)
pltref.set_xlabel('Time(second)')
pltref.set_ylabel('Template Strain')
pltref.set_title('Template')

fig.tight_layout()
plt.savefig('引力波', dpi=1000)
plt.show()
plt.close(fig)