フォルダ内のファイルを移動する繰り返し処理について

Question

pythonの繰り返し処理について質問させていただきます。
train_data_dir内のファイル7000個ずつを
tmp_dirのフォルダ10個へコピーするコードを作ろうとしています。
(1つのtrainフォルダ内のファイルをフォルダ10個に分割するコード)

import os
import shutil
from os import listdir
from shutil import copy
n = 1

train_data_dir="/content/drive/MyDrive/gamma_and_hadron/train/Gamma_train_70000/"
tmp_dir="/content/drive/MyDrive/tmp/gamma_tmp_"f'{n}'"/"

for counter in range(10):
 for file in listdir(train_data_dir)[:7000]: # フォルダ内の7000個という意味
     tmp_dir="/content/drive/MyDrive/tmp/gamma_tmp_"f'{n}'"/"
     copy(train_data_dir + file, tmp_dir)
     n = n+1
     if n == 11:
       break

Is a directory: '/content/drive/MyDrive/tmp/gamma_tmp_11/'
上記のエラーにより、途中で実行が止まってしまいます。
本来gamma_tmp_11が指定されないはずなのですが、どこを間違えているかがわかりません。
google colaboratory上で実行しています。
ご教示いただけますと幸いです。よろしくお願いいたします。

Comments

[y_waiwai]

このままではコードが読めないので、質問を編集し、</>(コードの挿入)ボタンを押し、出てくる’’’の枠の中にコードを貼り付けてください

[amachan4]

編集いたしました。
よろしくお願いいたします。

[bsdfan]

google colaboratoryとのことですが、n = 1 しているところと forのブロックを、違うセルに書いて、何度か実行していると、n = 11 で実行されているときがあるかもしれません。

Answer 1

shutil.copy はファイルしかコピーできず、ディレクトリをコピーするには shuril.copytree を使用しなければならなかったはずです。
なので、以下のようにコピー元のパスをファイルかディレクトリかを判定して、それぞれに対応した処理をすることでエラーは出なくなるかと思います。

import os
from shutil import copy, copytree

src_path = 'コピー元パス'
dist_path = 'コピー先パス'
if os.path.isfile(src_path):
    copy(src_path, dist_path)
else:
    copytree(src_path, dist_path)