【Python】if文を使ったファイル操作

動かしながら学ぶ PyTorchプログラミング入門

目次

問題

フォルダが100個あり、その中に2つの画像が入っているべきである。 しかし、フォルダによっては1つしかない場合がある。

目的

1つしか画像が入っていないフォルダのみを”mada”フォルダに集める。

コード

import os, glob
import shutil

if not os.path.isdir('mada'):
    os.makedirs('mada')

path = os.getcwd()  
# ファイルもフォルダも混ざったリスト
files = os.listdir(path)
# フォルダのみのリスト
dirs = [f for f in files if os.path.isdir(os.path.join(path, f))]

# フォルダごとにファイルの数を確認し、1であればmadaフォルダへコピー
for f in dirs:
    files = os.listdir(os.path.join(path, f))
    if len(files) == 1:
        shutil.copytree(f,("./mada/" + f))

頑張れ！、喝!!の代わりにB!ブックマークを押していただけるとただただうれしいです(^^)! ↓

動かしながら学ぶ PyTorchプログラミング入門