Help with a regex? (solved)

[wrybread]

I'm trying to isolate the number in a filename that can look like these:

DJI_0991.MP4
DJI_0992 - testing.MP4

The pattern is that it always starts with DJI_ and ends with either a period or a space.

Nevermind, solved! I don't know why this was giving me so much trouble. On the offchance anyone else comes this way:

import re

filename = "DJI_0991.MP4"

file_number = re.findall('DJI_(.+?)\W', filename)

print (file_number )

[deanhystad]

It probably gave you trouble because of the non-greedy matching. Another possibility is you might have used a control sequence where you wanted a backslash.

Why not grab all the digits instead of looking for the character after the digits?

import re

pattern = re.compile(r'DJI_(\d+)')

matches = re.findall(pattern, 'DJI_0991.MP4 DJI_0992 - testing.MP4 JI_0993 - testing.MP4 DJI 0994.MP4')
 
print(matches)

Output:
['0991', '0992']

[wrybread]

Because sometimes there's other digits in the filenames too, like "DJI_0090 - part 1.MP4"

[deanhystad]

I don't understand. In the example below should it print 0090 or not?

import re

pattern = re.compile(r'DJI_(\d+)')

matches = re.findall(pattern, 'DJI_0991.MP4 DJI_0992 - testing.MP4 DJI_0090 - part 1.MP4')
 
print (matches)

Output:
['0991', '0992', '0090']