import csv import os # 输入文件名和输出目录 csv_filename = "data.csv" output_dir = "output" # 创建输出目录 os.makedirs(output_dir, exist_ok=True) # 遍历 CSV 文件的每一行 with open(csv_filename, newline='', encoding='utf-8') as csvfile: reader = csv.DictReader(csvfile) for row in reader: # 提取标题和文本内容 title = row["title"] text = row["text"] # 创建新文件名,去掉标题中的非法字符 filename = "{}.txt".format(title.replace("/", "_")) # 将文本内容写入文件 with open(os.path.join(output_dir, filename), 'w', encoding='utf-8') as f: f.write(text) print("提取完成!")
时间: 2024-01-02 11:04:07 浏览: 89
这段代码的作用是读取一个名为"data.csv"的CSV文件,提取其中每一行的"title"和"text"字段。然后将"title"作为文件名,将"text"写入以"title"为文件名的文本文件中,并将文本文件保存在名为"output"的目录中。如果"title"中包含非法字符如"/",则将其替换为"_"。最后输出"提取完成!"。
阅读全文