Python怎么获取当前目录指定扩展名的文件名

Question

比如说我要获取所有.py文件的文件名，并让他显示文件名

Answer 1

import os
import os.path
ext=input('请输入你要显示的文件的扩展名（如.py）：')
print([f for f in os.listdir() if os.path.isfile(f) and os.path.splitext(f)[1]==ext])

Answer 2

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

if i.split(".")[-1] == "py":
print(i)本回答被网友采纳