python startswith正则表达式_python – 在列表中通过正则表达式过滤字符串

selected_files = filter(regex.match, files)

re.match('regex')等于re.search(‘^ regex’)或text.startswith(‘regex’)但是正则表达式版本.它只检查字符串是否以正则表达式开头.

所以,使用re.search()代替：

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 = list(filter(regex.search, files))

# The list call is only required in Python 3, since filter was changed to return a generator

print(selected_files)

输出：

['/a/b/c/la_seg_x005_y003.npy',

'/a/b/c/la_seg_x004_y003.npy',

'/a/b/c/la_seg_x003_y003.npy']

如果您只想获取所有.npy文件,只需使用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 = list(filter(lambda x: x.endswith('.npy'), files))

print(selected_files)