ディレクトリ内のファイル数を調べる

DIR = './dataset'

print(sum(os.path.isfile(os.path.join(DIR, name)) for name in os.listdir(DIR)))

最初、下のように書いていたのですが、@shiracamusさんに

リストを作るとメモリを大量消費するので、sum関数でbool値を合計するといいですよ。

とコメントいただきました。ありがとうございます！

↓メモリ大量消費するやつ。

DIR = './dataset'
print(len([name for name in os.listdir(DIR) if os.path.isfile(os.path.join(DIR, name))]))

2933

🐣

---

お仕事のご相談こちらまで
[e-mail]

Core MLを使ったアプリを作っています。
機械学習関連の情報を発信しています。

Twitter
Medium