在python中从两个不同的文件夹创建文本文件

时间:2015-05-04 22:59:18

标签: python

我正在尝试在python中编写一个包含两个不同文件夹(到文本文件)的列表;但是,当我运行我的程序时,只有我的第二个文件路径附加到列表并出现在我的文本文件中。如何附加我的列表,以便显示两个文件夹中的文件?我认为我的代码在第一步之后清空了列表。我想将这两个文件夹分开,并且不想仅仅为了列表而将它们组合起来。

import os
from os import listdir
from os.path import isfile, join

root = r'C:\GEO\Images\data\data_2008\ims_data_2008'
mylist = [ ]
for path, subdirs, files in os.walk(root):
     for name in files:
          mylist.append(os.path.join(path, name))

roottwo = r'C:\GEO\Images\data\data_2009\ims_data_2009'
mylist = [ ]
for path, subdirs, files in os.walk(roottwo):
     for name in files:
          mylist.append(os.path.join(path, name))

txt = open(r'C:\GEO6973\FinalExam\YoungResults\ims_data_list.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:\GEO6973\FinalExam\YoungResults\ims_data_list.txt', 'r') as f:
     read_data = f.read()

2 个答案:

答案 0 :(得分:2)

正如您所说,您的代码在第一个文件夹后清空列表。您必须在mylist = [ ]之后删除第二个roottwo,否则您将丢失到目前为止附加到列表中的所有数据。

在最后一个循环中,如果您尝试迭代附加到mylist的路径,请不要使用pop来访问其内容。代替:

for item in mylist:
    dir, filename = os.path.basename(os.path.dirname(item)), os.path.basename(item)
    txt.write(dir + ',' + filename + '\n')

最后,尽量避免使用list作为变量名,因为它是python中的保留关键字。

答案 1 :(得分:1)

设置rootwo后,您正在调用mylist = [ ]。这将创建一个新列表,有效地删除您之前附加到列表中的所有数据。