我正在尝试在python中读取excel文件,其中一些文件在表格前几行中包含信息。在表的列上是“可乐”。我想阅读发现“ cola”的文件。其他文件在第一行中具有列。当我运行以下代码时,它给我一个错误:“ Stopiteration”。我不知道我在做什么错。
import pandas as pd
import os
import itertools as it
#import shutil
rootdir = r"C:\\location"
for fname in os.listdir(rootdir):
file_path = os.path.join(rootdir,fname)
with open(file_path) as fp:
skip = next(it.ifilter(
lambda x: x[1].startswith('cola'),enumerate(fp)))[0]
if fname.endswith(".xlsx") or fname.endswith(".xls") or fname.endswith(".xlsm"):
print (os.path.join(rootdir,fname))
df = pd.read_excel(file_path, skiprows=skip)