【初心者向け】基礎&実践プログラミング

初心者がつまづきやすいところ、最短で実力が身につく方法をお伝えします。

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

f:id:AIProgrammer:20200419013021p:plain

問題

フォルダが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!ブックマークを押していただけるとただただうれしいです(^^)! ↓