时间戳

首页 > 解决方案 > 如何在一个文件夹中打开多个泡菜文件

python - 如何在一个文件夹中打开多个泡菜文件

问题描述

我在一个名为的文件夹中有多个格式相同的泡菜文件pickle_files：

1_my_work.pkl
2_my_work.pkl
...
125_my_work.pkl

我将如何将这些文件加载​​到工作区中，而不必一次执行一个文件？

谢谢！

标签： pythonloadpickle

解决方案

遍历文件并将数据保存在结构中，例如字典：

# Imports
import pickle
import os

# Folder containing your files
directory = 'C://folder'

# Create empty dictionary to save data
data = {}

# Loop over files and read pickles
for file in os.listdir(directory):
    if file.endswith('.pkl'):
        with open(file, 'rb') as f:
            data[file.split('.')[0]] = pickle.load(f)

# Now you can print 1_my_work
print(data['1_my_work'])

推荐阅读