如何将变量保存为for循环中的不同文件？

Question

我在列表中有一个csv文件路径名列表，我正在尝试将它们保存为数据框。我该怎么做？

import pandas as pd
import os
import glob

# use glob to get all the csv files
# in the folder
path = "/Users/azmath/Library/CloudStorage/OneDrive-Personal/Projects/LESA/2022 HY/All"
csv_files = glob.glob(os.path.join(path, "*.xlsx"))
  
# loop over the list of csv files
for f in csv_files:
    
    # read the csv file
    df = pd.read_excel(f)  
    display(df)
    print()

问题是它只打印。但我不知道如何保存。我想将所有数据框另存为变量，最好是它们的文件名。

Answer 1

试试这个：

a = [pd.read_excel(file) for file in csv_files]

然后，a 将是所有数据帧的列表。如果你想要字典而不是列表：

a = {file: pd.read_csv(file) for file in csv_files}

Answer 2

通过“保存”，我认为您的意思是将数据帧存储在变量中。我会为此使用字典而不是单独的变量。

import os


data = {}

for f in csv_files:
    name = os.path.basename(f)

    # read the csv file
    data[name] = pd.read_excel(f)  
    display(data[name])
    print()

现在，您的所有数据帧都存储在data字典中，您可以在其中迭代它们（如果需要，可以轻松地将所有数据帧一起处理）。字典中的键是输入文件的基本名称（文件名）。

回想一下，字典记住了插入顺序，所以文件的插入顺序也被保留。我可能建议在解析之前对输入文件进行排序——这样您就可以得到一个可复制的脚本和操作序列！