Найти все файлы, связанные с идентификатором, в папке с изображениями? ⇐ Python

[Anonymous]

Я пытаюсь заполнить фрейм данных на основе метки класса и изображений в папке.
У меня есть папка, содержащая более 10 000 изображений со следующей структурой имени: ['leaflet_10000_1.jpg', 'leaflet_10000_2.jpg', 'leaflet_10001_1.jpg', 'leaflet_10001_2.jpg', 'leaflet_10002_1.jpg', 'leaflet_10002_2.jpg', 'leaflet_10003_1.jpg', 'leaflet_10003_2.jpg'
И сопроводительный CSV-файл структуры:

Код: Выделить всё

ID,Location,Party,Representative/Candidate,Date
1000,Glasgow North,Liberal Democrats,,02-Apr-10
1001,Erith and Thamesmead,Labour Party,,02-Apr-10

Я хочу создать новый CSV-файл, содержащий пути ко всем изображениям указанной стороны. Я могу отделить определенную партию от полного файла csv с помощью команд:

Код: Выделить всё

df_ = df.loc[df["Party"] == "Labour Party"]

Это даст мне интересующую меня сторону, но как мне создать ПОЛНЫЙ список всех изображений, связанных с ней. Из приведенного выше списка изображений видно, что с идентификатором 1001 связано 2 изображения. Это не фиксированный номер, с некоторыми идентификаторами связано от 3 до 5 изображений.
Как мне получить этот новый фрейм данных, заполненный всеми необходимыми пути?
Я думаю применить str.split(name, '_') к каждому имени файла, а затем выполнить поиск по каждому идентификатору по всем результатам, но куда идти дальше?

Подробнее здесь: https://stackoverflow.com/questions/633 ... -of-images