在python中将文本文件名(类别)读入不同的类别字典？

category_dict = {}

import os

directory = '/path/to/directory'  # 替换为实际的目录路径

for filename in os.listdir(directory):
    # 处理每个文件名的逻辑

category = extract_category(filename)  # 根据实际情况提取类别信息

if category in category_dict:
    category_dict[category].append(filename)
else:
    category_dict[category] = [filename]

for category, filenames in category_dict.items():
    print(f"Category: {category}")
    print(f"Files: {filenames}")
    print()