Google Colabを用いてGoogle Drive内のフォルダを削除する

はじめに

Google colabでスクリプトを作る際には、Google Driveと連携させるのが便利です。
スクリプトにより生成されたファイルはGoogle Driveに保存されていきます。

連携において不便に感じるのは、ColabからDrive内のフォルダを削除しようとするとエラーが出ることです。
中身が空のフォルダは消せますが、中のファイルを1個1個消すのも面倒です。

Google Drive内のフォルダを削除するスクリプト

フォルダごと削除したいとき

shutil.rmtreeを使うと、指定したフォルダごと削除できます。
削除されたフォルダはGoogle Driveのごみ箱に入ります。

import shutil

directory = 'ここにパスをコピペ'

try:
    shutil.rmtree(directory)
except FileNotFoundError:
    pass

フォルダを残して中身だけ削除したいとき

中身ごと一度消してしまって、フォルダを作り直すと楽です。

import shutil

directory = 'ここにパスをコピペ'

try:
    shutil.rmtree(directory)
except FileNotFoundError:
    pass

#同名のフォルダを作成する
os.mkdir(directory)

下位のフォルダは残して、中のファイルのみ削除したいとき

glob.globを用いる方法もあります。

import os

directory = 'ここにパスをコピペ'
files = os.listdir(directory)
print(directory)
print(files)

for i in files:
    try:
        os.remove(directory+'/'+i)
    except IsADirectoryError:
        pass

おわりに

Colabのスクリプトの後ろにコピペしておくと、ちょくちょく掃除ができるので便利です。
では、快適なColab Lifeを！！