来自h5文件的引力波形图

时间:2016-02-07 23:59:09

标签: python python-3.x plot hdf5

我想绘制两个黑洞合并的波形我有一个.h5文件,我从公共波形目录中得到。我是一个使用python的初学者,所以我真的不知道在这种情况下该怎么做。我有一个.h5文件,里面有一个.dat文件,我想用来制作一个情节。我从公共波形目录中获取了该文件:

http://www.black-holes.org/waveforms/data/DisplayDownloadPage.php/?id=SXS:BBH:0001#

文件的名称是:

rhOverM_Asymptotic_GeometricUnits.h5

它位于Lev5目录中。 .h5文件的内容描述于:

https://www.black-holes.org/waveforms/docs.html

我认为文件中有一个数据集描述了我想绘制的波形。问题是我不知道如何进入数据集。我已经做到了:

import numpy as np
import h5py
from pylab import plot,show

f = h5.py.File("rhOverM_Asymptotic_GeometricUnits.h5","r")
ks = f.keys()

从这里开始,我不知道如何创建将进入绘图功能的x和y轴。我假设我需要另一个属于h5py模块的属性,但我甚至不确定我是否使用了正确的术语。任何帮助将不胜感激。

1 个答案:

答案 0 :(得分:2)

试试这个:

import matplotlib.pyplot as plt
import h5py

f = h5py.File("rhOverM_Asymptotic_GeometricUnits.h5", "r")
data = f['Extrapolated_N2.dir/Y_l2_m-1.dat']
plt.plot(data[:, 0], data[:, 1], label='column1')
plt.plot(data[:, 0], data[:, 2], label='column2')
plt.legend()
plt.show()