python - 如何使用变量名称列表从 .xls 文件夹自动创建 Pandas 数据框？

Question

我有一个包含许多文件的文件夹。我想将工作表 'sheet1' 作为它们每个的 pandas 数据框导入，并分配给该标题的一部分的名称。

我已成功获得文件名列表:

path = "/Users/path"
files = os.listdir(path)
files_xls = [f for f in files if f[-3:] == 'xls']

['A.xls', 'B.xls']

我已经成功地列出了我想要作为数据框名称的名称:

names = map(lambda each:each.strip(".xls"), files_xls)

['A', 'B']

但我无法使用这些名称进行导入。我可以手动完成:

A = pd.read_excel(A.xls, 'sheet1')
B = pd.read_excel(B.xls, 'sheet1')
etc...

但我不知道如何自动化这个过程。

Answer 1

使用.zip结合你的两个 lists变成 dict , 使用 .items() 迭代结果并收集pd.read_excel()在 dict :

df = {}
for name, f in dict(zip(names, files_xls)).items():
    df[name] = pd.read_excel(f, 'sheet1')

df['A']然后将包含 pd.read_excel('A.xls') 的结果.

您可以简化创建 list名称使用:

names = [f[:-4] for f in files_xls]

这工作如下:

files_xls = ['A.xls', 'B.xls']

names = [f[:-4] for f in files_xls]
['A', 'B']

name_dict = dict(zip(names, files_xls))
{'A': 'A.xls', 'B': 'B.xls'}

for name, f in name_dict.items():
    print(name, f)

打印:

A A.xls
B B.xls