目录名称+文件名到文本文件

时间:2012-06-20 15:03:08

标签: python file directory text-files

我有很多带有文件的目录。我想创建一个逗号分隔的txt文件,显示目录名称,以及该特定目录中的文件,请参阅下面的示例:

我在寻找什么:

DirName,Filename 
999,123.tif 
999,1234.tif 
999,abc.tif 
900,1236.tif 
900,xyz.tif 
...etc

下面的python代码将这些文件路径的列表推送到文本文件中,但是我不确定如何格式化列表,如上所述。

我目前的代码如下:

更新

我现在能够格式化文本文件,但是我注意到所有目录名/文件名都没有写入文本文件。它只写了~4000个dir /文件中的~4000个。是否有一些排序限制,我达到了文本文件行数,列表(mylist)大小,或一些阻止它的坏文件目录/文件字符(请参阅下面的更新代码)?

from os import listdir
from os.path import isfile, join
root = r'C:\temp'
mylist = ['2']
for path, subdirs, files in os.walk(root):
     for name in files:
          mylist.append(os.path.join(path, name))

txt = open(r'C:\temp\output.txt', 'w')
txt.write('dir' + ',' + 'file' + '\n')
for item in mylist:
     list = mylist.pop(0)
     dir, filename = os.path.basename(os.path.dirname(list)), os.path.basename(list)
     txt.write(dir + ',' + filename + '\n')

with open(r'C:\temp\output.txt', 'r') as f:
     read_data = f.read()

谢谢

1 个答案:

答案 0 :(得分:3)

也许这会有所帮助:

您可以获取绝对文件路径,然后执行以下操作

import os.path
p = "/tmp/999/123.tif"
dir, filename = os.path.basename(os.path.dirname(p)), os.path.basename(p)

结果:

In [21]: dir
Out[21]: '999'

In [22]: filename
Out[22]: '123.tif'

还可以考虑使用csv模块来编写此类文件。

import csv
import os.path

# You already have a list of absolute paths
files = ["/tmp/999/123.tif"]

# csv writer
with open('/tmp/out.csv', 'wb') as out_file:
    csv_writer = csv.writer(out_file, delimiter=',')
    csv_writer.writerow(('DirName','Filename'))
    for f in files:
        csv_writer.writerow((os.path.basename(os.path.dirname(f)), 
                             os.path.basename(f)))