Python - 我的输出在 CSV 中反转

Question

此程序扫描目录中的文件，并使用 REGEX（模式）提取数据。

我遇到的问题是输出如下：

程序的输出如下：

它打印正确，但 CSV 只需要一个条目，并且没有正确分配给列。 我想知道程序不会这样做

['EE-Data,DEC23,144009,TO,From,0000', 'FF-Data,DEC23,144009,TO,From,0000', 'GG-Data,DEC23,144009,TO,From,0000', 'HH-Data,DEC23,144009,TO,From,0000']
file2.TXT
['AA-Data,DEC23,144009,TO,From,0000', 'BB-DATA,DEC23,144509,TO,From,0000', 'CC-Data,DEC23,144009,TO,From,0000', 'DD-Data,DEC23,144009,TO,From,0000']
file1.TXT

但是在 CSV 中这样做

['AA-Data']['DEC23']['144009']['TO']['From']['0000']['File1.TXT']
['']['']['']['']['']['']['']
['']['']['']['']['']['']['']
['']['']['']['']['']['']['']
ETC 

如果我想添加更多模式，它们将进入每一行等。

# use regex on a directory of files and copy into a CSV
import re
import glob
import csv
import os


#Pattern REGEX configuration
patternCPECCC1 = "\/CC(.*?)\/C1"
patternCPET5S6 = "\/T5(.*?)\/S6"


path = "/test/"

#CSV headers
header = ['Column1', 'Column2', 'Column3', 'Column4', 'Column5', 'Column6', 'Column7'] # add more as I expand out the Patterns

# to search DIR defined in path
for files in glob.glob(path +"*.TXT"):
    infile = open(files)
    fin = infile.read()
    pat1 = re.findall(patternCPECCC1, fin)
##### to check what is coming from this for loop #########
    filename = os.path.basename(infile.name)
    print(pat1)
    print(filename)
######################################

with open('R.csv', 'w', newline='') as csvf:

    writer = csv.writer(csvf)
    writer.writerow(header)
    writer.writerow(pat1)
    writer.writerow([filename])

任何帮助将不胜感激，我已阅读 CSV 页面并了解其中的一些元素，我需要一些帮助来思考我的方法。谢谢

Answer 1

如果没有合适的.txt 文件，这有点难以测试，但您需要将脚本重组为如下内容：

# use regex on a directory of files and copy into a CSV
import re
import glob
import csv
import os
import io

#Pattern REGEX configuration
patternCPECCC1 = "\/CC(.*?)\/C1"
patternCPET5S6 = "\/T5(.*?)\/S6"
path = "/test/"

#CSV headers
header = ['Column1', 'Column2', 'Column3', 'Column4', 'Column5', 'Column6', 'Column7'] # add more as I expand out the Patterns

with open('R.csv', 'w', newline='') as csvf:
    writer = csv.writer(csvf)
    writer.writerow(header)
    
    # to search DIR defined in path
    for filename in glob.glob(path + "*.TXT"):
        with open(filename) as f_txt:
            data = f_txt.read()
            basename = f_txt.name
        
        pat1 = re.findall(patternCPECCC1, data)

        ##### to check what is coming from this for loop #########
        print(pat1)
        print(basename)
        ######################################

        rows = [next(csv.reader(io.StringIO(row))) + [basename] for row in pat1]
        writer.writerows(rows)

1. 在处理所有 .txt 文件时，您需要保持 CSV 文件处于打开状态。
2. pat1 的每个元素都是包含所有行的单个字符串。您需要使用 csv.reader() 将其拆分为值。这可以使用io.StringIO 来使字符串看起来像一个文件。
3. 您需要将文件名添加到每一行的末尾。
4. 然后您可以使用.writerows() 同时写入所有行