为什么os.remove（）或shutil.move（）只能移动部分文件

Question

我想从训练数据集中随机选择10个图像作为测试数据。如果我只将选定的数据复制到目标路径，它就可以工作。但是如果我想删除源数据，它只能删除其中的一些。我尝试了os.remove（）和shutil.move（）函数，但问题仍然存在。以下是我的剧本：

for label in labels:

    training_data_path_ch1 = os.path.join(training_data_folder, label, 'ch1')
    test_data_path_ch1 = os.path.join(test_data_folder, label, 'ch1')
    training_data_path_ch5 = os.path.join(training_data_folder, label, 'ch5')
    test_data_path_ch5 = os.path.join(test_data_folder, label, 'ch5')

    ch1_imgs = listdir(training_data_path_ch1)

    # Randomly select 10 images
    ch1_mask = np.random.choice(len(ch1_imgs), 10)
    ch1_selected_imgs = [ch1_imgs[i] for i in ch1_mask]

    for selected_img in ch1_selected_imgs:
        ch1_img_path = os.path.join(training_data_path_ch1, selected_img)
        shutil.copy2(ch1_img_path, test_data_path_ch1)
        os.remove(ch1_img_path)

    print('Successfully move ' + label + ' ch1 images')

我添加一张图片来显示运行状态。

你可以看到，该程序确实可以复制图像并删除一些图像，但为什么它不能删除所有图像？

有什么想法吗？我感谢任何帮助！

Answer 1

在：

ch1_mask = np.random.choice(len(ch1_imgs), 10)

您可能会多次返回相同的索引，这意味着您正在尝试复制已经复制和删除的文件（因此您无法在删除时再次复制该文件），而是通过{ {1}}，例如：

replace=False