Merge all file .pkl to one file .pkl Python3 (将所有文件合并为一个文件)

import os
import pickle

def merge_pkl_files(directory, output_file):
    merged_data = []
    for filename in os.listdir(directory):
        if filename.endswith('.pkl'):
            filepath = os.path.join(directory, filename)
            with open(filepath, 'rb') as file:
                data = pickle.load(file)
                merged_data.extend(data)
    
    # 保存合并后的数据到输出文件
    with open(output_file, 'wb') as file:
        pickle.dump(merged_data, file)

merge_pkl_files('/path/to/pkl/files', '/path/to/output/file.pkl')