|||
读hdf5文件和读fits差不多,需要知道类似文件头的东西,这里是一个变量的名称。这可以通过定义一个prt函数结合h5py报的工具来实现如下
def prt(name):
print(name)
filename=sys.argv[1]
f = h5py.File(filename, 'r')
f.visit(prt)
知道了变量名以后就可以读取相应的数据了。例如
vx=np.array(f['velocity_x'])
之后就和通常的操作一样了。总的读取数据画图的脚本如下。
import h5py
import sys
import numpy as np
import matplotlib.pyplot as plt
import matplotlib
matplotlib.style.use('classic')
def prt(name):
print(name)
filename=sys.argv[1]
f = h5py.File(filename, 'r')
f.visit(prt)
vx=np.array(f['velocity_x'])
integrated_vx=np.sum(vx,axis=0)
plt.imshow(integrated_vx)
plt.show()
Archiver|手机版|科学网 ( 京ICP备07017567号-12 )
GMT+8, 2024-9-26 14:20
Powered by ScienceNet.cn
Copyright © 2007- 中国科学报社