Использование PdfFileMerger в Python для объединения PDF с одинаковыми именами, но разными номерами

У меня есть каталог с отдельными PDF-файлами, которые нужно объединить, основываясь на их названии. Каждый отдельный PDF-файл содержит одну страницу. Система именования для каждого файла состоит из строкового имени и номера. Вот как примерно выглядит мой каталог:

A_001.pdf
A_002.pdf
A_003.pdf
B_001.pdf
B_002.pdf
B_003.pdf
B_004.pdf

В основном мне нужен один PDF для A (PDF будет содержать 3 страницы) и один PDF для B (PDF будет содержать 4 страницы). Числа _001 и так далее должны быть номером страницы. Мой текущий Python-скрипт генерирует A.pdf и B.pdf, но включает страницы как из A, так и из B.

import PyPDF2, os
from PyPDF2 import PdfFileReader, PdfFileWriter, PdfFileMerger
from pathlib import Path

single_file_dir = r'Y:\Python\Single_PDFs'
binder_file_dir = r'Y:\Python\Combined_PDFs'

# получить список всех файлов в каталоге с отдельными PDF
single_file_list = []
for file in os.listdir(single_file_dir):
    if file.endswith(".pdf"):
        single_file_list.append(single_file_dir + "\\" + file)

print(single_file_list)

# собрать имена файлов для выходных многостраничных PDF

file_name_list = []
for file in single_file_list:
    name = os.path.basename(file)
    new_name = name[:-8]
    file_name_list.append(new_name)
    unique_file_name_list = list(set(file_name_list))

merger = PdfFileMerger()

print(unique_file_name_list)

# попытаться сопоставить имя входного файла с именем выходного файла
for file in single_file_list:
    for name in unique_file_name_list:
        if name in file:
            merger.append(file)
            merger.write(binder_file_dir + "\\" + name + ".pdf")

Этот скрипт действительно генерирует A.pdf и B.pdf, но оба выходных PDF включают множество дубликатов как отдельных PDF A, так и отдельных PDF B. Моя цель — объединить A_001.pdf, A_002.pdf, A_003.pdf в один многостраничный PDF. То же самое с PDF-сериалами B.

Я думаю, что ваша проблема может быть связана с повторным использованием вашего PDF-объединителя.

Этот код адаптирован из другого скрипта, который я использую для объединения PDF. Дайте мне знать, если он работает для вас.

from collections import defaultdict
from pathlib import Path

from PyPDF2 import PdfMerger

single_file_dir = Path("Y:/") / "Python" / "Single_PDFs"
binder_file_dir = Path("Y:/") / "Python" / "Combined_PDFs"

file_groups: defaultdict[str, list[Path]] = defaultdict(list)
for file in single_file_dir.glob("*.pdf"):
    group = file.name[0]  # Как вы хотите определить группу по имени файла
    file_groups[group].append(file)

for group, files in file_groups.items():
    merger = PdfMerger()
    for file in sorted(files):
        merger.append(file)

    with open(binder_file_dir / f"{group}.pdf", "wb") as binder:
        merger.write(binder)

Примечания:

Мне нравится использовать модуль pathlib, чтобы избежать работы с специфическими для платформы особенностями путей (особенно \‘s на Windows)