p72

[Mudwan7]

5.4.1例子中应该是return[f for f in os.listdir(path) if f.endswith(ext)] file错了对吧

[wingfish]

是的，你说得对，是错了，我重新验证了一下，应该是返回变量 f

import os


def list_files_with_ext(path, ext):
    return [f for f in os.listdir(path) if f.endswith(ext)]


print(os.listdir('/Users/*****/Documents/dev_python'))

print(list_files_with_ext('/Users/*****/Documents/dev_python', 'py'))

非常感谢！