python - フィルタとマップを使用して、Pythonで拡張子が.txtおよび.pyのディレクトリ内のすべてのファイルを検索します

Question

フィルターとマップ機能を使用して、.txtまたはPythonで終わるディレクトリ内のファイルを検索するにはどうすればよいですか?.py

score 5 · Accepted Answer

os.listdirファイルのリストが表示されます。True次に、ファイル名が or で終わるときに.py戻る関数を書くだけです.txt:

filter(lambda x: x.endswith(('.txt','.py')), os.listdir(os.curdir))

…どうやって組み込めばいいのかわからないmap…

score 2 · Accepted Answer

次の 2 つの方法があります。

純粋な機能的アプローチ:

from operator import methodcaller
filter(methodcaller('endswith', ('.txt', '.py')), os.listdir('.'))

リスト理解アプローチ:

[fn for fn in os.listdir('.') if fn.endswith(('.txt', '.py'))]

お役に立てれば：-）

score 0 · Accepted Answer

glob モジュールを使用してこれらのファイルを取得できます

>>> import glob    
>>> glob.glob('*.py')
['test.py', 'a.py', 'b.py']

4 に答える 4