有选择地将文件从一个文件夹目录复制到另一个文件夹

Question

我有一个目录树，其中文件夹的名称很重要。我也有一个csv说从folder1> folder2> folder3> foo.txt。 folder1，folder2，folder3和txt都位于csv的不同列中。我需要保持目录结构不变，并复制csv中提供的文件。

我正在尝试的方法是复制目录树并编写python代码以删除不需要的文件。因此，有很多循环，但是我在csv中有超过415,000行。

csv example:<br/>
pdf_no .   folder1. folder2 . folder3. <br/> 1 .  . abc. pqr. xyz.<br/>

这是csv的格式，借助python中的pandas dataframe提取列数据没有问题。最初，这是一个.dta文件，我将其转换为.csv with pandas. So 'folder1' > 'folder 2' > 'folder 3' > 'pdf_no'.“ pdf_no”。列中包含文件名，这是我们想要在给定文件夹中保持文件结构的数字。

因此需要很多时间，每当我再次更改时，都会花费很多时间，我什至不知道它是否正确。

Answer 1

csv示例

pdf_no,folder1,folder2,folder3
1,abc,def,ghi
2,xyz,pqr,
3,abc,def,ghi

示例代码

import csv
import os
import shutil


target_csv = 'selection.csv'
target_dir = 'selected_20190828/'
source_dir = 'original_directory/'

with open(target_csv) as f:
    rows = csv.reader(f)
    for line_no, row in enumerate(rows):
        if line_no == 0:  # Skip the first line because it's the title
            continue

        pdf_name = row[0] + '.pdf'
        dir_path = os.path.join(*row[1:])

        source = os.path.join(source_dir, dir_path, pdf_name)
        if not os.path.isfile(source):
            print('not exist: ', line_no, source)
            continue
        target = os.path.join(target_dir, dir_path)
        os.makedirs(target)
        shutil.copy2(source, target)

说明

您实际上不需要pandas，您只需

• csv.reader将CSV文件读入list
• os.makedirs创建文件夹（此方法类似于bash中的mkdir -p）
• os.path.join
• shutil.copy2将文件复制到新文件夹
• os.path.isfile以确保原始文件存在

我已经测试了上面的代码。应该可以。

Answer 2

您需要使用shutil.copytree方法。这是您可以做的：

1. 阅读您的CSV
2. 使用os.path.join（）构建文件路径
3. 使用shutil.copytree将文件及其父目录复制到目标

在目标文件已经存在时，也许您必须添加try...except块来避免OsError，或者在复制新文件之前删除目标文件。