如何从泡菜文件列表中创建多个熊猫数据帧？

Question

我想为每个泡菜文件列表创建一个熊猫数据框。

如果我使用此代码：

dfs = [pd.read_pickle(file) for file in DATA_PROCESSED.iterdir()]

我将得到一个列表，但我也想在文件名和数据框名称之间建立对应关系。在这种情况下，我只会得到dfs [0]

你能帮我吗？

Answer 1

您可以使用字典理解而不是列表理解。您可以从字典键中删除扩展名，以便以后使用.strip('.pkl')轻松访问，用您正在使用的任何扩展名替换.pkl。

dfs = {file.strip('.pkl'):pd.read_pickle(file) for file in DATA_PROCESSED.iterdir()}

然后，您可以使用常规词典语法访问每个变量：

dfs['filename']