ディレクトリ内に異なる拡張子を持つファイルが混ざっていているとし(↓のような状況)、どの拡張子がいくつあるのか数えるようなコードを作成してみます。
my_directory/
├── file1.html
├── file2.txt
├── file2.html
└── sub_folder
├── file3.html
今回は、試しにmy_directoryの.html
のファイル数を数えてみましょう。
import os
からos.listdir(directory)
↓のような関数を定めます。
import os
def count_html_files(directory):
# ディレクトリ内のすべてのファイルをリスト化
files = os.listdir(directory)
# HTMLファイルのみを抽出してカウント
html_files = [file for file in files if file.endswith('.html')]
return len(html_files)
そして、さっき作成した関数を用いて以下のコードを実行すれば完了です!
count_html_files(任意のディレクトリ名)
ポイント ~files = os.listdir(directory)
~
files
には何がどのように格納されているでしょうか?
files = ['file1.html', 'file2.txt', 'file2.html', 'sub_folder', 'file3.html']
各ファイルがリスト形式で格納されているのですね。