Python解析HDF文件

谁借莪１个温暖的怀抱￠ 2022-08-01 12:29 203阅读 0赞

前段时间因为一个业务的需求需要解析一个HDF格式的文件。在这之前也不知道到底什么是HDF文件。百度百科的解释如下：

> HDF是用于存储和分发科学数据的一种自我描述、多对象文件格式。HDF是由美国国家超级计算应用中心NCSA(全称：National Center for Supercomputing Application）创建的，为了满足各种领域研究需求而研制的一种能高效存储和分发科学数据的新型数据格式。HDF可以表示出科学数据存储和分布的许多必要条件。

使用Python解析当然会用到第三方的包，如下：

import math
    import pandas as pd
    import xlwt

第一个是用来做数学计算的`math包主要处理数学相关的运算`。至于关于`pandas`的介绍请[点击这里][Link 1]。xlwt这个包是写HDF文件的。

使用Python读取HDF文件的代码如下：

with closing(pd.HDFStore(HDF_FILR_URL)) as store:
            df = store[date]
    
        # index shoule be end -> region -> group
        df.reset_index(inplace=True)
        df.set_index(["end", "region", "group"], inplace=True)
        df.sort_index(inplace=True)

其实这样获取到数据之后就是`pandas`提供的函数，获取自己需要的数据。

slice_df = df.loc[dt]
        rtt = slice_df.rtt.unstack(level=0) / 1000
        cwnd = slice_df.cwnd.unstack(level=0)
        total = slice_df.total.unstack(level=0)
        rows = rtt.index.tolist()
        columns = rtt.columns.tolist()

最后写入Excel，代码如下：

def writexcel(listname, name, time):
        #将数据写入Excel
        saveurl = EXCEL_FILR_URL + '%s_%s_%s.xls' % (AVG_RTT, time, name)
        excel_file = xlwt.Workbook()
        table = excel_file.add_sheet('tcpinfo')
    
        index_row = 0
        for item in listname:
            for item_key, item_value in item.items():
                table.write(index_row, 0, str(item_key))
                table.write(index_row, 1, str(item_value[1][0]))
                table.write(index_row, 2, str(item_value[1][1]))
                table.write(index_row, 3, str(item_value[0]).decode('utf-8'))
                index_row += 1
    
        excel_file.save(saveurl)

[Link 1]: http://cloga.info/python/%E6%95%B0%E6%8D%AE%E7%A7%91%E5%AD%A6/2013/09/17/pandas_intro/