Python load指定目录下指定后缀的文件

1）指定文件的路径

path = "D:\\Python\\PyCharm_Projects\\tensorflow-DBN-master"

实际上在windows上的路径是一个反斜杠的：D:\Python\PyCharm_Projects\tensorflow-DBN-master
要注意的是反斜杠转义的问题，这里使用前一个“\”取消第二个”\”的转义作用，或者说这种方式表示一个反斜杠。
一般正斜杠“/”在linux中可以作为目录的分隔符，但是我在windows上使用的时候报错了。

2）使用os.path模块提供的os.listdir() 方法获取目录下的完整文件名

import os
dir = os.listdir(path)

3）使用os.path模块提供的os.splitext() 方法选择指定的文件
也可通过这个方法获取文件的后缀名。

for x in dir:                             
    if os.path.splitext(x)[1] == ".npy":   
        print(x)  

4）使用numpy的load()函数加载npy文件

import numpy as np
dataset = np.load("/路径/XX.npy")

这里路径的拼接如下：

filePath = path + "\\" + x

5) 完整示例

import numpy as np
import os
dataset = {} #使用字典保存数据
path = "D:\\Python\\PyCharm_Projects\\tensorflow-DBN-master"
dirs = os.listdir(path)
for x in dirs:
    if os.path.splitext(x)[1] == ".npy":
        filePath = path + '\\' + x        #拼接完整路径
        data_key = os.path.splitext(x)[0] #去除文件后缀，获取文件名
        data_value = np.load(filePath)    #载入数据    
        dataset[data_key] = data_value    #字典的键值对应

key_list = dataset.keys() #将字典的键以列表的形式表示，可以查看键