python 批量处理

Question

我有一堆csv文件，每一个系列从kmeansN01__1到kmeansN01__180，有很多N01、N02等。我想求一个代码，可以根据文件名字中N01这样的序号自动生成一个文件夹，把他们序号下的文件输入到该文件夹，同时文件夹名称以该系列的N01、N02、N03等

这样命名

Answer 1

基于Monster 组和GPT的调写：
可以根据文件名中的序号自动生成文件夹，将对应的文件移动到文件夹中：

import os
import shutil

# 指定csv文件所在目录
csv_dir = '/path/to/csv/directory'

# 获取目录下所有的csv文件
csv_files = [f for f in os.listdir(csv_dir) if f.endswith('.csv')]

for csv_file in csv_files:
    # 获取csv文件名
    file_name = os.path.splitext(csv_file)[0]
    # 获取序号
    num = file_name.split('__')[1].split('_')[0]

    # 创建文件夹
    folder_name = f'N{num}'
    folder_path = os.path.join(csv_dir, folder_name)
    if not os.path.exists(folder_path):
        os.mkdir(folder_path)

    # 移动文件到文件夹
    file_path = os.path.join(csv_dir, csv_file)
    new_file_path = os.path.join(folder_path, csv_file)
    shutil.move(file_path, new_file_path)

将获取目录下所有以.csv结尾的文件，对于每个文件，它将从文件名中提取序号（例如，从kmeansN01__1.csv中提取01），然后创建一个名为N01的文件夹，并将具有相应序号的所有文件移动到该文件夹中。如果文件夹已经存在，则跳过创建步骤。