用Pandas设置标题

Question

我有一些txt文件，他们开始了大量的BS和20至30行后，有用的部分开始。我想用数字前面的最后一行作为标题。我知道如果我知道确切的行号，我可以设置为我的标题（使用pd.read\U csv文件)但是对于每个文件，这个数字是不同的（正如我说的，它在20到30之间）。我知道我要找的台词是以“潜力”开头的。有什么简单的使用方法吗pd.read\U 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)