使用Python从具有独特头打印技术的文件中解析头字符串（.inp扩展名）

Question

我希望使用Python来解析文件中的数据帧（对于那些可能使用过它的SWMM模型input/inp文件的人）。文件头以一种非常独特的方式打印，这使得完全解析它变得非常困难。我试图从文件中读取的数据帧的示例是：

;;                                                 Param    
;;Node           Parameter        Time Series      Type     
;;-------------- ---------------- ---------------- -------- 
80408            FLOW             80408            FLOW     
81009            FLOW             81009            FLOW     
82309            FLOW             82309            FLOW     

标题没有用制表符或任何固定数量的空格分隔。另外，对于某些页眉，当它们的长度太大时，它们会占用两条垂直线，而其他标头只使用一条直线。宽度也不是固定的，有不止一个这样的数据帧，它们的宽度都不同。在

我所能做的就是抓住最下面的那条线作为头球。在

^{pr2}$

Answer 1

与@tdelaney的代码相同的思想使其更加简洁：

from itertools import takewhile
import re
import pandas as pd

def make_dataframe(filename):
    with open(filename) as fp:
        div = []
        headers = list(takewhile(
            lambda x: not x.startswith(';;-') or div.append(x), fp))
        colspecs = [m.span() for m in re.compile("-+").finditer(div[0])]
        headers = [
            re.sub(" +", " ", " ".join(hl[lo : hi] for hl in headers)).strip()
            for lo, hi in colspecs]
        d = pd.read_fwf(fp, header = None,
            colspecs = [(lo - 2, hi - 2) for lo, hi in colspecs])
    d.columns = headers
    return d

print(make_dataframe('test.txt'))