带有文件名的几个文本文件到 csv

Question

我在一个文件夹中有几个文本文件，我想将它们转换为 csv。在每一行中，我想在一列中包含我的文件名，在另一列中包含内容，例如：

file1.txt，该文件中包含的所有文本

file2.txt，这里有更多文字

...

我正在使用以下代码：

from glob import glob
import pandas as pd
files=os.listdir("where the text files are saved")
content=[]
for file in files: 
    with open(file, 'r', newline='') as source_file: 
        for line in csv.reader(source_file): 
            content.append(line[0]) 
with open('output.csv', 'w') as target_file: 
    for line in content: 
        target_file.write(out_filename + "," + line + '\n')

但是，使用此代码，我只能在 csv 中获取文本文件的内容。谁能解释我如何在列中添加文件名？

Answer 1

import csv
import pandas as pd
filelocation = "FILELOCATION"
files= [k for k in next(os.walk(f"{filelocation}"))][2]
filenames = []
content=[]
for file in files:
    with open(f"{filelocation}\\{file}", 'r', newline='') as source_file:
        filenames.append(file)
        lines = source_file.read()
        content.append(lines)

data = {'Filename':filenames,
        'Content':content}

df = pd.DataFrame(data)
print (df)

df.to_csv(r'Path where you want to store the exported CSV file\File Name.csv', index = False)