用熊猫设置标题

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)