python 特定のフォルダの削除

Question

フォルダXの中に、
フォルダa1,a2,a3,a4,,,,,a99,a100があり、
それぞれのフォルダの中にs,t,u,vというフォルダがあるとします。

a1～a100までのフォルダ内のsまたはtをフォルダ名に含むフォルダを一括削除するには、どのようなコードを書けばよいのでしょうか？

aが頭につくフォルダ全ての中身を正規表現で一括操作したいと思っています。

folder = 'X'
for inputfilename in os.listdir(folder) :
    for input in os.listdir(folder + '/' + inputfilename) :

この続きがさっぱりわかりません。

質問の仕方があまりよくなかったので修正しました。

Answer 1

itertools.productを使ってみたかったんです！！！

# -*- coding: utf8 -*-
import itertools
from pathlib import Path
import shutil
folder = 'X'


def main():
    target_dir = [Path(*p) for p in itertools.product(folder, 'abcd', 'st')]
    for subdir in target_dir:
        print(subdir)
        shutil.rmtree(str(subdir))


def test_create_dir():
    for p in itertools.product(folder, 'abcd', 'stuv'):
        subdir = Path(*p)
        subdir.mkdir(parents=True, exist_ok=True)
        #print(subdir)


if __name__ == '__main__':
    #test_create_dir()
    main()

---

2018/02/05追記
ご要望の正規表現ではありませんが、改訂された質問内容に対応する削除処理を記載しておきます。

# -*- coding: utf8 -*-
from itertools import starmap, product, repeat
from pathlib import Path
import shutil
folder = 'X'


def sub_dir() -> map:
    return map(''.join, zip(repeat('a'), map(str, range(101))))


def main() -> None:
    for p in starmap(Path, product(folder, sub_dir(), 'st')):
        print(p)
        #shutil.rmtree(str(p))


def test_create_dir() -> None:
    for p in starmap(Path, product(folder, sub_dir(), 'stuv')):
        p.mkdir(parents=True, exist_ok=True)


if __name__ == '__main__':
    # テストのためのディレクトリ作成！
    #test_create_dir()
    main()

Comments

[YouheiSakurai]

賛同します。starmap(Path, product(...))やmap(Path, *product(...))でもカッコいいですね。

[umyu]

starmap初めて知りました。。回答文に反映させていただきました。参考になりました。ありがとうございます。

Answer 2

難しく考えずにrmtreeを繰り返したらいいんです。

from shutil import rmtree


rmtree('X/a/s')
rmtree('X/a/t')
rmtree('X/b/s')
rmtree('X/b/t')
rmtree('X/c/s')
rmtree('X/c/t')
rmtree('X/d/s')
rmtree('X/d/t')

これが冗長だと感じたらそこからちょっと難しく考え出せば良いのです。

追記:質問変更にに対応

正規表現を使うと難しくなるのでglobをお勧めします。

from glob import glob

for target in glob('X/a*/s') + glob('X/a*/t'):
    print(target)

Comments

[grx72expo]

回答ありがとうございます。
print(target)を
shutil.rmtree(target)にして、目的の処理を実行できました。