如何在循环中动态创建目标文件名列表？

Question

我正在尝试以.csv格式创建其中具有特定字符串的文件列表，但是尽管循环会打印每个文件名，但我的列表仅存储所有文件名中的最后一个。

我试过创建一个列表，然后将其输出为csv，但输出的唯一内容是列表的最后一个元素。

for fname in glob.glob('*.txt'):   
  if os.path.isfile(fname):    
    with open(fname) as f:   
        for line in f:       
            if 'target' in line:    
                mylist = []
                mylist.append(fname)
                #print ('found code in file %s' %fname)
                print(mylist)
                with open("out.csv","w") as l:
                    wr = csv.writer(l,delimiter="\n")
                    wr.writerow(mylist)
                    break

此代码的输出是

['target_1.txt']
['target_3.txt']

我希望这是csv格式，但是当我查看out.csv文件时，该文件中只有target_3.txt。 我想要的是带有行的csv：

['target_1.txt']
['target_3.txt']

Answer 1

以@JonClements的评论并发布作为答案，可以更轻松地理解他在说什么。

with open("out.csv","w") as l:                # Open "out.csv" ONCE
    for fname in glob.glob('*.txt'):   
        if os.path.isfile(fname):    
            with open(fname) as f:   
                for line in f:       
                    if 'target' in line:    
                        mylist = []
                        mylist.append(fname)
                        #print ('found code in file %s' %fname)
                        print(mylist)
                        wr = csv.writer(l,delimiter="\n")
                        wr.writerows(mylist)
                        break

Answer 2

注意缩进的区别。 与其在for循环内执行第二步with不如在循环完成后以相同的缩进级别执行。

mylist = []
for fname in glob.glob('*.txt'):
  if os.path.isfile(fname):
    with open(fname) as f:
        for line in f:
            if 'target' in line:
                mylist.append(fname)
                #print ('found code in file %s' %fname)
                break
with open("out.csv","w") as l:
    wr = csv.writer(l,delimiter="\n")
    wr.writerows(mylist)

还要注意我们如何在for循环之前创建mylist ； 您将在循环内覆盖该列表的先前值（也是如此）。 如评论中所述，我还将writerow更改为writerows以写入所有收集的行。