用熊猫设置标题

Question

我有一些 txt 文件，它们以很多 BS 开头，在 20 到 30 行之后，有用的部分开始了。我想使用数字之前的最后一行作为我的标题。我知道如果我知道确切的行号，我可以将其设置为我的标题（使用 pd.read_csv），但对于每个文件，该数字是不同的（正如我所说的在 20 到 30 之间）。我知道我正在寻找的线路以“潜力”开头。有什么简单的方法可以使用 pd.read_csv 并从头开始设置标题。

Answer 1

您可以使用“传统”文件 I/O 方法读取文件顶部并计算行数，直到找到标题行。知道它的编号后，使用pandas.read_csv() 重新读取文件。

with open(yourfile) as infile:
    for n,row in enumerate(infile):
        if row.startswith("Potential"):
            break

df = pd.read_csv(yourfile, skiprows=n)