ファイル名のみの一覧を取得する際のプログラミングについて

Question

初心者です。
pythonで、ファイル名のみの一覧を取得する際のプログラミングについて教えてください。
ネットで調べたところ、以下のようなコードを利用できるかと思いました。

ファイル構造の中身は、以下のような感じです。
.
└── testdir
├── dir1
├── dir2
├── file1
├── file2.txt
└── file3.jpg

import os
path = "./testdir"
files = os.listdir(path)
files_file = [f for f in files if os.path.isfile(os.path.join(path, f))]
print(files_file)

[f for f in files if os.path.isfile(os.path.join(path, f))]の処理手順がわかっていません。

forの前に、fが入るケースがそもそもわからず、どのようなプログラムになっているのかわからないので

かみ砕いてご教授いただけますでしょうか。

よろしくお願いいたします。

Comments

[meg_]

質問のタイトルと質問内容がずれていると思います。質問タイトルの「ファイル名のみの一覧、取得」については解決済なのでしょうか？

[star999]

失礼いたしました。

タイトル名を変更しました。実行はうまくできたのですが、以下のロジックが理解できていません。

[f for f in files if os.path.isfile(os.path.join(path, f))]

よろしくお願いいたします。

Answer 1

これは、「リスト内包表記」というものです。

以下のようなプログラムと同値といえます。

files_file = []
for f in files:
    if os.path.isfile(os.path.join(path, f)): 
        files_file.append(f)

リストの内容を短く記述できます。また、速度が若干早いという利点もあります。

f　について

[f for f in files]

この場合、 files から一つずつ要素を取り出して、そのまま新しいリストの要素とします。

[str(f) for f in files]

この場合は files から取り出した要素を str() に渡して、結果をリストの要素とします。
ゆえにリストは、フォルダ名やファイル名の文字列を含みます。

つまり、 forより前に書かれている f はリストの要素を示しています。

なお、リストに入れる要素を示しているだけなので、 f を使わなくてもエラーにはなりません：

[1 for f in files]

この場合は [1, 1, 1, ...] というリストになりますね

if 以下について

もし条件式を評価して、 True ならその要素を新しいリストの要素とし、 False なら無視するという動きをします。

os.path.isfile(os.path.join(path, f))

os.path.isfile は与えられたパスがファイルであれば、True を返します。なので、

[f for f in files if os.path.isfile(os.path.join(path, f))]

は、files の要素を f として、もしそのパスがファイルであれば、 f を新しいリストの要素にする、という動きをします。

Answer 2

内包表記です。

Pythonリスト内包表記の使い方をお読みください。

公式ドキュメントでは5.1.3. リストの内包表記に書いてあります。