想要在 python 中使用 os.walk 获得所需的图像路径

Question

我有名为 A 的文件夹，其中包括一些以字母 A 开头的子文件夹。在这些子文件夹中放置了不同的图像（一些图像格式是 .png、jpeg、.giff 和 .webp，具有不同的名称，如 item1 .png、item1.jpeg、item2.png、item3.png 等）。 从这些子文件夹中，我想获取以 1 结尾的图像的路径列表。除此之外，我只想获得 1 种图像文件格式，例如仅用于 jpeg。 在某些子文件夹中，图像名称以 1.png、1.jpeg、1.giff 等结尾。我只想要每个以 1 结尾的子文件夹中的一个图像。（任何图像格式）。 我正在共享返回所有图像格式的项目图像路径（以 1 结尾）的代码。

代码：

Answer 1

这是可以解决您的问题的代码。

import os
img_paths = []
for top,dirs, files in os.walk("your_path_goes_here"):
    for pics in files:
        if os.path.splitext(pics)[0] == '1':
            img_paths.append(os.path.join(top,pics))
            break
        

img_paths将拥有您需要的列表

它将具有来自子文件夹的第一个图像，名称为1 ，可以是任何格式。

如果您想要特定格式，

import os
img_paths = []
for top,dirs, files in os.walk("your_path_goes_here"):
    for pics in files:
        if os.path.splitext(pics)[0] == '1' and os.path.splitext(pics)[1][1:] in ['png','jpg','tif']:
            img_paths.append(os.path.join(top,pics))
            break

感谢S3DEV使其更加优化