Python 数据分析与可视化

”引力波的绘制“实例绘制：

程序源码：
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, hstrain = wavfile.read(r’L1_Strain.wav’, “rb”)

reftime, ref_H1 = np.genfromtxt(‘GW150914_4_NR_waveform_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 = hstrain.shape[0] / rate_l
ltime = np.arange(-ltime_len / 2, ltime_len / 2, ltime_interval)

fig = plt.figure(figsize=(12, 6)) # 创建一个大小为12*6的绘图空间

plth = fig.add_subplot(221)
plth.plot(htime, hstrain, ‘y’)

plth.set_xlabel(‘Time(seconds’) #画出以时间为x轴，应变数据为y轴的图像，并设置标题和坐标轴的标签
plth.set_ylabel(‘H1 Strain’)
plth.set_title(‘H1 Strain’)

plth = fig.add_subplot(222)
plth.plot(ltime, hstrain, ‘g’)

plth.set_xlabel(‘Time(seconds’) # 画出以时间为x轴，应变数据为y轴的图像，并设置标题和坐标轴的标签
plth.set_ylabel(‘L1 Strain’)
plth.set_title(‘L1 Strain’)

plth = fig.add_subplot(212)
plth.plot(reftime, ref_H1)

plth.set_xlabel(‘Time(seconds’) # 画出以时间为x轴，应变数据为y轴的图像，并设置标题和坐标轴的标签
plth.set_ylabel(‘L1 Strain’)
plth.set_title(‘L1 Strain’)
fig.tight_layout()

fig.tight_layout()

plt.savefig(‘Gravitational_Waves_Original.png’)
plt.show()
plt.close(fig)

配置文件在：www.python123.org
要将其中的txt文件该名成：GW150914_4_NR_waveform_template.txt。