Python将每个子目录中的文件数输出到csv文件

时间:2012-11-21 22:55:35

标签: python csv

我试图将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)

2 个答案:

答案 0 :(得分:4)

你应该尝试以下方面:

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

打印子目录和文件数。

答案 1 :(得分:0)

我不清楚你的问题,你可能想重新阅读os.walk文档。 root是要遍历的当前目录。 dirsroot中的子目录,files是直接位于root中的文件。现在您的代码计算相同的文件(从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])