列表理解中的匹配、过滤和分组

Question

我有一个文件夹，其中包含以下格式的一些图像：

wheel_0.jpg, tyre_2.jpg

还有其他一些格式：

bar_0.heic

和这样的文件（我不想匹配）：

hello.jpg

我想创建一个 jpeg 格式的图像名称列表，不要以 _0 结束并添加它们而不添加它们的扩展名。 我已经制作了这段可以正常工作的代码：

images = os.listdir("images")
images_to_search = []
for image in images:
    re_obj = re.search("(.+)(_\d+)(\..+)", image)
    if re_obj:
        if re_obj.group(3) == ".jpg" and re_obj.group(2) != "_0":
            images_to_search.append(re.sub("\.jpg", '', image))

有什么办法可以将这个 for loo 变成列表理解？

Answer 1

您可以使用

import re

images = os.listdir("images")
rx = re.compile(r'.+_(?!0\.)\d+\.jpg$')
images_to_search = [x.rsplit('.')[0] for x in filter(rx.match, images)]
# => ['tyre_2']

请参阅正则表达式演示。 正则表达式匹配

• .+ - 除换行符以外的任意一个或多个字符，尽可能多
• _(?!0\\.)\\d+ - _后面不跟0.然后是一位或多位数字
• \\.jpg - .jpg文本。

re.match要求匹配只发生在字符串的开头，不需要在模式前加上^ 。

请参阅Python 演示：

images = ['tyre_2.jpg', 'bar_0.heic', 'hello.jpg', 'wheel_0.jpg']
import re
rx = re.compile(r'.+_(?!0\.)\d+\.jpg$')
print([x.rsplit('.')[0] for x in filter(rx.match, images)])
# => ['tyre_2']

Answer 2

我认为你可以在不使用re情况下做到这一点，就像这样

>>> [f.replace(".jpg","") for f in images if "_" in f and f.endswith(".jpg") and not f.endswith("_0.jpg")]

Answer 3

images = os.listdir('images')
images_to_search = [image for image in images if re.search("(.+)(_\d+)(\..+)", image).group(3) == '.jpg' and re.search("(.+)(_\d+)(\..+)", image).group(2) != '_0')

Answer 4

你可以试试这个：

images = [x[:-4] for x in images if (x[-4:]==".jpg")&(x[-4:]!="_0.jpg")]