1

我正在尝试将 Linux 系统上每个子目录中的文件数汇总到 Excel 表中。

目录一般设置：maindir/person/task/somedata/files. 但是，设置的子目录会有所不同（即，某些文件可能没有 ' task' 目录），所以我需要让 python 遍历文件路径。

我的问题是我需要从“ person”开始的所有子目录名称，目前我的代码（如下）仅将最近的目录与文件计数附加在一起。如果有人能帮我解决这个问题，将不胜感激！

import os, sys, csv

outwriter = csv.writer(open("Subject_Task_Count.csv", 'w'))

dir_count=[]
os.chdir('./../../')
rootDir = "./" # set the directory you want to start from
for root, dirs, files in os.walk( rootDir ):
for d in dirs:
    a = str(d)
    count = 0
    for f in files:
        count+=1
    y= (a,count)
    dir_count.append(y)

for i in dir_count:
    outwriter.writerow(i)
4

2 回答 2

4

您应该尝试以下方式：

for root,dirs,files in os.walk( rootDir ) :
    print root, len(files)

它打印子目录和文件数。

于 2012-11-21T23:01:03.090 回答
0

我不清楚你的问题，你可能想重新阅读 os.walk 文档。 root是被遍历的当前目录。 dirs是 中的子目录root，files是 中的文件root。由于您的代码现在是您计算相同的文件（从根目录）并将其记录为每个子目录中的文件数。

这就是我想出的。希望它接近你想要的。如果没有，请调整 :) 它会打印目录、目录中的文件数以及目录及其所有子目录中的文件数。

import os
import csv

# Open the csv and write headers.
with open("Subject_Task_Count.csv",'wb') as out:
    outwriter = csv.writer(out)
    outwriter.writerow(['Directory','FilesInDir','FilesIncludingSubdirs'])

    # Track total number of files in each subdirectory by absolute path
    totals = {}

    # topdown=False iterates lowest level (leaf) subdirectories first.
    # This way I can collect grand totals of files per subdirectory.
    for path,dirs,files in os.walk('.',topdown=False):
        files_in_current_directory = len(files)

        # Start with the files in the current directory and compute a
        # total for all subdirectories, which will be in the `totals`
        # dictionary already due to topdown=False.
        files_including_subdirs = files_in_current_directory
        for d in dirs:
            fullpath = os.path.abspath(os.path.join(path,d))

            # On my Windows system, Junctions weren't included in os.walk,
            # but would show up in the subdirectory list.  this try skips
            # them because they won't be in the totals dictionary.
            try:
                files_including_subdirs += totals[fullpath]
            except KeyError as e:
                print 'KeyError: {} may be symlink/junction'.format(e)

        totals[os.path.abspath(path)] = files_including_subdirs
        outwriter.writerow([path,files_in_current_directory,files_including_subdirs])
于 2012-11-21T23:44:04.430 回答