ciągi filtrów według regex na liście

Question

Chciałbym filtrować listę ciągów w Pythona za pomocą regex. W następującym przypadku przechowuj tylko pliki z rozszerzeniem ".npy".

Kod, który nie działa:

import re 

files = [ '/a/b/c/la_seg_x005_y003.png', 
      '/a/b/c/la_seg_x005_y003.npy', 
      '/a/b/c/la_seg_x004_y003.png', 
      '/a/b/c/la_seg_x004_y003.npy', 
      '/a/b/c/la_seg_x003_y003.png', 
      '/a/b/c/la_seg_x003_y003.npy', ] 

regex = re.compile(r'_x\d+_y\d+\.npy') 

selected_files = filter(regex.match, files) 
print(selected_files) 

To samo wyrażenie regularne działa na mnie w Ruby:

selected = files.select { |f| f =~ /_x\d+_y\d+\.npy/ } 

Co jest nie tak z kodem Pythona?

Answer 1

selected_files = filter(regex.match, files) 

re.match('regex') jest równa re.search('^regex') lub text.startswith('regex') ale wersji regex. Sprawdza tylko, czy ciąg rozpoczyna się od regex.

więc używać re.search() zamiast:

import re 

files = [ '/a/b/c/la_seg_x005_y003.png', 
      '/a/b/c/la_seg_x005_y003.npy', 
      '/a/b/c/la_seg_x004_y003.png', 
      '/a/b/c/la_seg_x004_y003.npy', 
      '/a/b/c/la_seg_x003_y003.png', 
      '/a/b/c/la_seg_x003_y003.npy', ] 

regex = re.compile(r'_x\d+_y\d+\.npy') 

selected_files = filter(regex.search, files) 
print(selected_files) 

wyjściowa:

['/a/b/c/la_seg_x005_y003.npy', 
'/a/b/c/la_seg_x004_y003.npy', 
'/a/b/c/la_seg_x003_y003.npy'] 

---

A jeśli po prostu chcesz uzyskać wszystkie .npy plików, wystarczy użyć str.endswith():

files = [ '/a/b/c/la_seg_x005_y003.png', 
      '/a/b/c/la_seg_x005_y003.npy', 
      '/a/b/c/la_seg_x004_y003.png', 
      '/a/b/c/la_seg_x004_y003.npy', 
      '/a/b/c/la_seg_x003_y003.png', 
      '/a/b/c/la_seg_x003_y003.npy', ] 


selected_files = filter(lambda x: x.endswith('.npy'), files) 

print(selected_files) 

Answer 2

Po prostu użyj search - od dopasowania zaczyna się od początku do końca (tj. Całego) ciągu znaków i dopasowań wyszukiwania w dowolnym miejscu ciągu.

import re 

files = [ '/a/b/c/la_seg_x005_y003.png', 
      '/a/b/c/la_seg_x005_y003.npy', 
      '/a/b/c/la_seg_x004_y003.png', 
      '/a/b/c/la_seg_x004_y003.npy', 
      '/a/b/c/la_seg_x003_y003.png', 
      '/a/b/c/la_seg_x003_y003.npy', ] 

regex = re.compile(r'_x\d+_y\d+\.npy') 

selected_files = filter(regex.search, files) 
print(selected_files) 

Output-

['/a/b/c/la_seg_x005_y003.npy', '/a/b/c/la_seg_x004_y003.npy', '/a/b/c/la_seg_x003_y003.npy'] 

Answer 3

Jeśli match, wzór musi obejmować całą wejście . Albo przedłużyć Ci wyrażenia regularnego:

regex = re.compile(r'.*_x\d+_y\d+\.npy') 

Które pasuje:

['/a/b/c/la_seg_x005_y003.npy', 
'/a/b/c/la_seg_x004_y003.npy', 
'/a/b/c/la_seg_x003_y003.npy'] 

Albo użyć re.search, co

skanów poprzez ciąg szukasz pierwszym miejscu gdzie wyrażenie regularne wzór tworzy dopasowanie [...]

Answer 4

szuka dopasowania na początku łańcucha. Zamiast tego możesz użyć re.search().