0

实际上我有一个文件夹（下图中的数据），其中包含 2 个子文件夹，每个子文件夹都包含一些 .png 文件。我需要遍历每个子文件夹并对该子文件夹中的每个图像文件进行一些编码并保存结果。我用过os.walk()，os.listdir()但glob.glob()没有一个有效。我尝试过的许多代码之一与以下相同：

path1 = Path('./data')
path2 = os.listdir(path1)

# loop through main folder to read each subfolder
for i in path2:
    if not i.startswith('.'):
       path3 = Path(os.path.join(path1,i))
       path4 = os.listdir(path3)

    #loop through each subfolder to read each file
       for j in path4:
           #some coding

在此处输入图像描述

任何建议将不胜感激。

4

4 回答 4

1

我建议使用pathlib库。该库是一个“面向对象的文件系统路径”模块，它结合了 Python 最好的文件系统模块，如 os、os.path 和 glob。

from pathlib import Path

path1 = Path('./data')
files = [item.as_posix() for item in path1 .glob('**/*.png') if item.is_file()]

这将为您提供数据子文件夹中所有 .png 路径的列表。

于 2019-03-04T19:54:47.880 回答
0

你可以listdir()这样使用：

# pathname of root dir
images_path = "./data"

# filtered file extension
suffix = ".png"

# For each image,
for i in os.listdir(images_path):
    file = os.path.basename(i)
    fileName, fileExtension = os.path.splitext(file)
    # is it an image file with 'suffix' extension ?
    if os.path.isfile(images_path+'/'+i) and fileExtension == suffix:
        # do some coding
于 2019-03-04T19:55:44.857 回答
0

我可以找到我的答案！这很简单，但我在命名时犯了一个错误。因此，下面编写的代码可能会帮助其他人遇到同样的问题：

path = "./data/"

for subfolder in os.listdir(path):
    subfolder_name = path + subfolder

    for imgs in os.listdir(subfolder_name):
        imagename = subfolder_name + '/' + imgs

        # do some coding
于 2019-03-05T15:03:28.863 回答
0

像这样的东西os.walk：

import os
for root, dirs, files in os.walk(path_to_data_folder):
#    if not root.endswith(good_folder_name):
#        continue
    for fname in files:
        if fname_meets_my_criteria:
            fpath = os.path.join(root, fname)
            with open(fpath, 'r') as f, open(new_file_path, 'w') as newfile:
                data = f.read()
                # process file data
                new_data = func_that_processes_data(data)
                newfile.write(new_data)

这有一些伪代码：

  • fname_meets_my_criteria是比较的替代品，如果您想过滤文件进行处理，它需要这个 - 它可能类似于fname.edswith('.txt')或not fname.endswith('.cfg')

  • new_file_path是将要写入已处理数据的新文件的路径和名称。

如果您打算在处理完文件后覆盖文件，请改用：

for root, dirs, files in os.walk(path_to_data_folder):
#    if not root.endswith(good_folder_name):
#        continue
    for fname in files:
        if fname_meets_my_criteria:
            fpath = os.path.join(root, fname)
            with open(fpath, 'r') as f:
                data = f.read()
            # process file data
            new_data = func_that_processes_data(data)
            with open(fpath, 'w') as f:
                f.write(new_data)

在我的两个示例中，文件都是作为文本文件打开的。如果您需要处理字节而不是测试/字符串，请打开带有mode参数的文件'rb'或'wb'

于 2019-03-04T19:59:38.280 回答