通過所有輸入文件循環任務python

Question

我想要統計所有.txt文件中的所有As和Bs和Cs，並提供一個.csv文件來列出所有這些字母中的一個。

這裏的代碼做了我想要的，但只有最後一個文件我提供，而不是所有的人。

我在做什麼錯？

import glob 
import csv 

#This will print out all files loaded in the same directory and print them out 
for filename in glob.glob('*.txt*'): 
    print(filename) 

#A B and C 
substringA = "A" 
Head1 = (open(filename, 'r').read().count(substringA)) 
substringB = "B" 
Head2 = (open(filename, 'r').read().count(substringB)) 
substringC = "C" 
Head3 = (open(filename, 'r').read().count(substringC)) 
header = ("File", "A Counts" ,"B Counts" ,"C Counts") 
analyzed = (filename, Head1, Head2, Head3) 

#This will write a file named Analyzed.csv 
with open('Analyzed.csv', 'w', newline='') as csvfile: 
    writer = csv.writer(csvfile) 
    writer.writerow(header) 
    writer.writerow(analyzed) 

Answer 1

還有另一個小改變你需要做的：你需要打開作爲追加，不寫，以及縮進。請注意，當您以追加方式打開時，您將不會覆蓋之前存在的任何內容，因此我在頂部添加了部分以刪除已在csv中的任何內容。

import glob 
import csv 


#This will delete anything in Analzyed.csv if it exists and replace it with the header 
with open('Analyzed.csv','w') as csvfile: 
    writer = csv.writer(csvfile) 
    header = ("File", "A Counts" ,"B Counts" ,"C Counts") 
    writer.writerow(header) 

for filename in glob.glob('*.txt*'): 
    print(filename) 

    #A B and C 
    substringA = "A" 
    Head1 = (open(filename, 'r').read().count(substringA)) 
    substringB = "B" 
    Head2 = (open(filename, 'r').read().count(substringB)) 
    substringC = "C" 
    Head3 = (open(filename, 'r').read().count(substringC)) 
    header = ("File", "A Counts" ,"B Counts" ,"C Counts") 
    analyzed = (filename, Head1, Head2, Head3) 

    #This will write a file named Analyzed.csv 
    with open('Analyzed.csv', 'a', newline='') as csvfile: 
     writer = csv.writer(csvfile) 
     writer.writerow(analyzed) 

以上是我的解決方案，儘可能保持儘可能多的代碼不變。不過，理想情況下，您只能在文件的開頭打開一次文件。這是你將如何做到這一點：

import glob 
import csv 


with open('Analyzed.csv','w') as csvfile: 
    writer = csv.writer(csvfile) 
    header = ("File", "A Counts" ,"B Counts" ,"C Counts") 
    writer.writerow(header) 

    for filename in glob.glob('*.txt*'): 
     print(filename) 

     #A B and C 
     substringA = "A" 
     Head1 = (open(filename, 'r').read().count(substringA)) 
     substringB = "B" 
     Head2 = (open(filename, 'r').read().count(substringB)) 
     substringC = "C" 
     Head3 = (open(filename, 'r').read().count(substringC)) 
     analyzed = (filename, Head1, Head2, Head3) 

     writer.writerow(analyzed) 

Answer 2

壓痕失蹤，追加模式a開放Analyzed.csv：

import glob 
import csv 

#This will print out all files loaded in the same directory and print them out 
for filename in glob.glob('*.txt*'): 
    print(filename) 

    #A B and C 
    substringA = "A" 
    Head1 = (open(filename, 'r').read().count(substringA)) 
    substringB = "B" 
    Head2 = (open(filename, 'r').read().count(substringB)) 
    substringC = "C" 
    Head3 = (open(filename, 'r').read().count(substringC)) 
    header = ("File", "A Counts" ,"B Counts" ,"C Counts") 
    analyzed = (filename, Head1, Head2, Head3) 

    #This will write a file named Analyzed.csv 
    with open('Analyzed.csv', 'a') as csvfile: 
     writer = csv.writer(csvfile) 
     writer.writerow(header) 
     writer.writerow(analyzed) 

編輯：刪除不支持newline=""參數

Answer 3

你可以試試這個：

from itertools import chain 
from collections import Counter 
for filename in glob.glob('*.txt*'): 
    data = chain.from_iterable([list(i.strip("\n")) for i in open(filename)]) 

    the_count = Counter(data) 
    with open('Analyzed.csv', 'w', newline='') as csvfile: 
     writer = csv.writer(csvfile) 
     writer.writerow(filename) 
     writer.writerow("A count: {}".format(the_count["A"])) 
     writer.writerow("B count: {}".format(the_count["B"])) 
     writer.writerow("C count: {}".format(the_count["C"]))