如何将单独的 Excel 工作表读入单独的 DataFrame？

Question

我有一个包含 13 个选项卡的 Excel 文件，我想编写一个函数，从文件中获取指定的工作表，将它们转换为单独的 DataFrame，然后将它们捆绑到 DataFrame 列表中。在这种情况下，我想获取标有“tblProviderDetails”、“tblSubmissionStatus”和“Data Validation Ref Data”的表格，将它们转换为 DataFrames 并制作一个列表。我希望将 dfs 放在列表中的原因是，我最终想要获取输入的 dfs 并返回一个字典，然后该字典将用于创建 YAML 文件。

这最终是我想要的：

dfs = [ 'tblProviderDetails', 'tblSubmissionStatus', 'Data Validation Ref Data']

我想使用用户定义函数的原因是我希望灵活地将任何工作表和任何数量的工作表调用到列表中。

我能够编写一个将单个指定工作表转换为数据框的函数，但我不确定如何在 Excel 文件中调用任意数量的工作表或在函数中创建一个列表。这是据我所知：

def read_excel(path, sheet_name, header):
    dfs = pd.read_excel(path, sheet_name=sheet_name, header=header)
    return dfs

df1 = read_excel(path=BASEDIR, sheet_name='tblProviderDetails', header=2)
df2 = read_excel(path=BASEDIR, sheet_name='tblSubmissionStatus', header=2)
df3 = read_excel(path=BASEDIR, sheet_name='Data Validation Ref Data', header=2)

感谢您的帮助。

Answer 1

有多种方法可以做到这一点，但也许最简单的方法是首先获取所有工作表名称，然后在每个工作表名称的循环中，将结果加载到数据框中并将其附加到所需的列表中。

dfList = []
def read_excel(path, h):
    xls = pd.ExcelFile(path)

    # Now you can access all sheetnames in the file
    sheetsList = xls.sheet_names
    # ['sheet1', 'sheet2', ...]

    for sheet in sheetsList:
        dfList.append(pd.read_excel(path, sheet_name=sheet, header 
        =h))
read_excel('book.xlsx',2)
print(dfList)

Answer 2

您可以将工作表名称和\或工作表编号的列表传递给参数 sheet_name。

def read_excel(path, sheet_name, header):
    sheet_name = ['tblProviderDetails','tblSubmissionStatus','Data Validation 
    Ref Data']
    dfs = pd.read_excel(path, sheet_name=sheet_name, header=header)
    return dfs