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("提取完成！")

这段代码的作用是读取一个名为"data.csv"的CSV文件，提取其中每一行的"title"和"text"字段。然后将"title"作为文件名，将"text"写入以"title"为文件名的文本文件中，并将文本文件保存在名为"output"的目录中。如果"title"中包含非法字符如"/"，则将其替换为"_"。最后输出"提取完成！"。

相关问题

import os import pydicom from PIL import Image # 将某个 IMA 文件转换为 PNG 格式并保存 def save_as_png(ds, png_path): # 将 IMA 像素数据转换为图像 pixel_data = ds.pixel_array image = Image.fromarray(pixel_data) image = image.convert('L') # 灰度化 # 保存为 PNG 格式 image.save(png_path) # 定义输入和输出目录 input_dir = './data' output_dir = './output' # 遍历每个子目录 for subdir in os.listdir(input_dir): subdir_path = os.path.join(input_dir, subdir) if not os.path.isdir(subdir_path): continue print(f'Processing {subdir}') # 在输出目录中创建相同的子目录 out_subdir_path = os.path.join(output_dir, subdir) os.makedirs(out_subdir_path, exist_ok=True) # 遍历 full_3mm 和 quarter_3mm 文件夹 for mode in ['full_3mm', 'quarter_3mm']: mode_path = os.path.join(subdir_path, mode) if not os.path.isdir(mode_path): continue # 遍历每个 IMA 文件并转换为 PNG 格式 for filename in os.listdir(mode_path): if not filename.endswith('.ima'): continue ima_path = os.path.join(mode_path, filename) ds = pydicom.dcmread(ima_path) # png_filename = f'{subdir}_{mode}_{ds.SOPInstanceUID}.png' # 使用所有信息作为文件名 png_filename = os.path.splitext(filename)[0] + '.png' png_path = os.path.join(out_subdir_path, png_filename) save_as_png(ds, png_path)为什么无生成

这段代码可能无法生成PNG文件是因为文件保存路径不存在或没有写入权限。你需要检查输出目录是否存在，如果不存在则需要使用`os.makedirs()`创建目录。同时，你也需要检查代码中使用的输出文件名是否正确，如果文件名中包含非法字符或长度过长，可能会导致文件无法保存。你可以添加一些调试代码，例如调用`os.path.exists()`检查文件路径是否存在，以及添加异常处理代码，以便在保存失败时输出错误信息。

if file == file_name: output_subdirectory = os.path.join(output_directory, subdirectory) os.makedirs(output_subdirectory, exist_ok=True) # 创建输出子文件夹 source_file = os.path.join(subdirectory_path, file) destination_file = os.path.join(output_subdirectory, file) shutil.copy(source_file, destination_file) print(f"已复制文件: {file} 到目录: {output_subdirectory}")

很抱歉，我之前的回答有误。在你的代码中，`file == file_name`这个条件是错误的，因为`file`是文件名，而`file_name`是不包括扩展名的文件名。你可以使用`os.path.splitext()`函数来获取文件名和扩展名，然后比较文件名是否相同。

以下是修正后的代码示例：

import os
import shutil

def get_files_with_same_name(directory, output_directory):
    # 获取目录下的所有文件夹
    subdirectories = [name for name in os.listdir(directory) if os.path.isdir(os.path.join(directory, name))]

    for subdirectory in subdirectories:
        # 构建当前子文件夹的完整路径
        subdirectory_path = os.path.join(directory, subdirectory)

        # 获取当前子文件夹下的所有文件
        files = [name for name in os.listdir(subdirectory_path) if os.path.isfile(os.path.join(subdirectory_path, name))]

        # 使用字典来存储相同文件名的文件列表
        files_dict = {}
        for file in files:
            file_name = os.path.splitext(file)[0]  # 只获取文件名，不包括扩展名
            if file_name not in files_dict:
                files_dict[file_name] = []
            files_dict[file_name].append(file)

        # 将相同文件名的文件复制到输出目录中
        for file_list in files_dict.values():
            if len(file_list) > 1:
                output_subdirectory = os.path.join(output_directory, subdirectory)
                os.makedirs(output_subdirectory, exist_ok=True)  # 创建输出子文件夹

                for file in file_list:
                    source_file = os.path.join(subdirectory_path, file)
                    destination_file = os.path.join(output_subdirectory, file)
                    shutil.copy(source_file, destination_file)
                    print(f"已复制文件: {file} 到目录: {output_subdirectory}")

# 指定目录和输出目录
directory = 'path/to/directory'
output_directory = 'path/to/output_directory'

# 调用函数获取相同文件名的文件并保存到指定文件夹
get_files_with_same_name(directory, output_directory)

在修正后的代码中，我们将`file`的比较改为比较文件名（不包括扩展名），并遍历字典的值来获取相同文件名的文件列表。然后，将这些文件复制到指定的输出目录中的子文件夹中。

请确保在运行代码之前已经安装了必要的模块（os、shutil），并将目录路径替换为你自己的实际路径。