如何在 pandas 中使用 read_excel 提高处理速度？

Question

我需要使用 pd.read_excel 来处理一个 excel 文件中的每一张纸。
但在大多数情况下，我不知道工作表名称。
所以我用这个来判断excel中有多少张：

i_sheet_count=0
i=0
try:
  df.read_excel('/tmp/1.xlsx',sheetname=i)
  i_sheet_count+=1
  i+=1
else:
  i+=1
print(i_sheet_count)

在这个过程中，我发现这个过程很慢，
那么，read_excel能否只读取有限的行来提高速度？
我尝试了 nrows 但没有成功..仍然很慢..

Answer 1

阅读所有工作表而不用猜测

对pd.read_excel 使用sheetname = None 参数。这会将所有工作表读入数据框字典。例如：

dfs = pd.read_excel('file.xlsx', sheetname=None)

# access 'Sheet1' worksheet
res = dfs['Sheet1']

限制行数或列数

您可以使用parse_cols 和skip_footer 参数来限制列数和/或行数。这将减少阅读时间，也适用于sheetname = None。

例如，以下将读取前 3 列，如果您的工作表有 100 行，它将只读取前 20 行。

df = pd.read_excel('file.xlsx', sheetname=None, parse_cols='A:C', skip_footer=80)

如果您希望应用特定于工作表的逻辑，可以通过提取工作表名称来实现：

sheet_names = pd.ExcelFile('file.xlsx', on_demand=True).sheet_names

dfs = {}
for sheet in sheet_names:
    dfs[sheet] = pd.read_excel('file.xlsx', sheet)

提高性能

将 Excel 文件读入 Pandas 自然比其他选项（CSV、Pickle、HDF5）要慢。如果您希望提高性能，我强烈建议您考虑这些其他格式。

例如，一个选项是use a VBA script 将您的 Excel 工作表转换为 CSV 文件；然后使用pd.read_csv。