如何在python中的文件夹中获取具有特定扩展名的文件？

Question

我正在研究一个python代码。每次在主文件夹中生成新文件夹（包含带扩展名为npy的文件）时，我都会更改运行代码的几个参数。我想访问新文件夹中的所有npy文件。如果我使用

`os.path.listdir()`

它只列出主文件夹中的文件。如何处理扩展名为npy的文件？

Answer 1

您需要在函数中指定目录的路径

致电时：

`os.path.listdir()` 

默认情况下会转到当前目录。例如，如果您的新文件夹名为foobar，那么您需要将其写为：

path = "/foobar/"
os.listdir( path )

或等效地应也

os.listdir( "/foobar/" )

如果您需要任何澄清，请告诉我

修改

如果您只想要具有npy扩展名的文件，可以使用以下代码：

for file in os.listdir("/foobar"):
if file.endswith(".npy"):
    print(os.path.join("/npy", file))

再次，请告诉我你是否需要澄清。