旅行好きなソフトエンジニアの備忘録

プログラミングや技術関連のメモを始めました

【Python】 フォルダ内の特定のファイルを取得する

C#のDirectory.GetFilesに相当するPythonの関数を探したのでそのメモ。

フォルダ内のファイル名一覧を取得する

import os

files = os.listdir('Folder\\')
for file_name in files:
    print(file_name)

実行結果例
a.txt
b.txt
c.txt
d.docx

フォルダ内のテキストファイル名のみ取得する(方法1)

import os

files = os.listdir('Folder\\')
extension = '.txt'
text_files = [file for file in files if extension in file]
for file_name in text_files:
    print(file_name)

実行結果例
a.txt
b.txt
c.txt

フォルダ内のテキストファイル名のみ取得する(方法2)

import glob

extension = '.txt'
text_files = glob.glob('Folder\\*' + extension)
for file_path in text_files:
    print(file_path)

実行結果例(中途半端にパスが付いてきます)
Folder\a.txt
Folder\b.txt
Folder\c.txt