python获取指定目录下所有文件名列表的方法？

写段伪代码import os# 获取当前目录path = os.getcwd()# 获取当前目录所有文件列表list_dir = os.listdir(path)for i in list_dir: # 字符串分割，判断"."后面是否为"py" if i.split(".")[-1] == "py": print(i)