利用python.os.listdir获取文件夹中所有文件名

利用内置os的listdir可以获取文件夹中的文件名，从而可以利用文件名对数据做进一步的处理，比如：在猫狗识别的练习过程中，需要通过文件名对猫狗数据进行分类。

def get_file(file_dir):
    dogs = []
    cats = []
    dog_labels = []
    cat_labels = []
    #利用listdir可以获取文件夹中所有文件的名称
    for dir in os.listdir(file_dir):
        temp = dir.split(".")
        if temp[0] == 'dog':
            dogs.append(file_dir+dir)
            dog_labels.append(1)
        else:
            cats.append(file_dir+dir)
            cat_labels.append(0)