Как выбирать файл в правильной поочередности с названием от нуля до максимального?

Question

[bybacapitan]

Есть функция, которая создает видео из кадров

spoiler

def createvideo():
	img_array = []
	for filename in glob.glob('result/*.png'):
		img = cv2.imread(filename)
		height, width, layers = img.shape
		size = (width,height)
		img_array.append(img)


	out = cv2.VideoWriter('project.avi',cv2.VideoWriter_fourcc(*'DIVX'), 15, size)
	
	for i in range(len(img_array)):
		out.write(img_array[i])
	out.release()

в кажом названии кадра присутсвует цифра.

spoiler

Если я запускаю эту фунцию, то кадры склеиваются рандомно.
Что нужно сделать для того чтобы кадры склеивались в правильной поочередности?

Comments

[Влад Григорьев]

отсортировать полученные данные, извлекая числа и строки и сортируя по ним

Answer 1

[PavelMos]

Если бы в названиях была бы сохранена разрядность, то есть слева дописаны нули, то можно было бы отсортировать сразу по названиям. Но тут надо извлекать цифры с помощью регекспа, делать дополнительный отсортированный список или словарь и проходить уже по нему. Шаблон регекспа для имён типа файл1, файл20, файл200. Если в полном пути файла, если его создаёт glob.glob(...), могут также цифры встречаться, надо шаблон другой делать.

import re
l=['frame1.jpg','frame2.jpg','frame3.jpg','frame13.jpg','frame100.jpg','frame101.jpg','frame200.jpg']
#l = glob.glob('result/*.png'):
r='(\d{1,3})'
out={int(re.search(r,x).group(0)):x for x in l}
for number in sorted(out.keys()):
    filename=out[number]
    img = cv2.imread(filename)
    height, width, layers = img.shape
    size = (width,height)
    img_array.append(img)

Comments

[bybacapitan]

название могу изменить, потому что сам его же и задаю, когда разбиваю видео на кадры

[bybacapitan]

AttributeError: 'NoneType' object has no attribute 'group'

[PavelMos]

bybacapitan,
можно ли при задании шаблонов названий делать так, чтобы нули слева дописывались ? Тогда сильно проще будет.

Ошибка с group значит шаблон не ищется в имени файла. Странно, у меня ищется. Выложи на всякий случай кусок списка файлов из glob.glob(....

l=['frame1.jpg','frame2.jpg','frame3.jpg','frame13.jpg','frame100.jpg','frame101.jpg','frame200.jpg']
print ('\r\n'.join(sorted(l))) #обычная сортировка по алфавиту
#l = glob.glob('result/*.png'):
r='(\d{1,3})'
out={int(re.search(r,x).group(0)):x for x in l}
for k, v in out.items(): print (v, '(',k,')') #сортировка по извлеченным цифрам
frame1.jpg
frame100.jpg
frame101.jpg
frame13.jpg
frame2.jpg
frame200.jpg
frame3.jpg
frame1.jpg ( 1 )
frame2.jpg ( 2 )
frame3.jpg ( 3 )
frame13.jpg ( 13 )
frame100.jpg ( 100 )
frame101.jpg ( 101 )
frame200.jpg ( 200 )

[bybacapitan]

PavelMos, всмысле нули?
типо frame01.png ?

[bybacapitan]

PavelMos, Я же вроде правильно сделал

spoiler

async def createvideo():
	img_array = []
	l = glob.glob('result/*.png')
	r = '(\d{1,3})'
	out={int(re.search(r,x).group(0)):x for x in l}
	for number in sorted(out.keys()):
		filename=out[number]
		img = cv2.imread(filename)
		height, width, layers = img.shape
		size = (width,height)
		img_array.append(img)


	out = cv2.VideoWriter('project.avi',cv2.VideoWriter_fourcc(*'DIVX'), 15, size)

	for i in range(len(img_array)):
		out.write(img_array[i])
		out.release()

[PavelMos]

bybacapitan, да.
Если так будет, то обычная сортировка т.е. sorted (список) должна будет их правильно расставить, или же они через glob.glob(...) будут сразу же выданы в правильном порядке (если конечно в директории других файлов не будет)

[bybacapitan]

PavelMos,

Теперь выглядит так, куда нужно добавить sorted()?

[PavelMos]

bybacapitan, подумой сам.
У тебя есть некий список элементов, который проходится с помощью конструкции FOR elemenint IN spisok..
Надо, чтобы эти элементы обрабатывались поочерёдно по нужному порядку (сортировка). Значит, можно сделать проход по уже отсортированному списку. А так как в Питоне всё это объекты и к ним можно применять функции, то

for filename in sorted(glob.glob('result/*.png')):
   img = cv2.imread(filename)
   height, width, layers = img.shape
   size = (width,height)
   img_array.append(img)
.....

Answer 2

[mayton2019]

Нужно определить какой будет самый максимальный кадр. Ну допустим это 9999. Ну и все остальные попереименовывать к чорту типа frame1.png => frame0001.png.

Для сисадмина-девопса задачка на 5 минут.