1.Python os.listdir() 方发

语法

os.listdir(path)

参数

path – 需要列出的目录路径

返回值

返回指定路径下的文件和文件夹列表。

实例

#!/usr/bin/python
# -*- coding: UTF-8 -*-

import os, sys

# 打开文件
path = "/var/www/html/"
dirs = os.listdir( path )

# 输出所有文件和文件夹
for file in dirs:
   print file

复制

2.Python读取文件夹下的所有文件工程实例

import os
import cv2
import numpy as np
from paths import PATHS

img_path = 'D:\\Anomly_Dataset\\UCSD_Anomaly_Dataset.v1p2\\UCSDped1\\Train' 
#文件夹目录
fileNames = os.listdir(img_path)  # 获取当前路径下的文件名，返回List
img_list = [] 

for file in fileNames:      #遍历文件夹
    img_folder = img_path + '\\' + file  #文件中的文件名
    print(img_folder)
    if os.path.exists(img_folder):  #判断是否存在该文件

        for image_name in os.listdir(img_folder):  #遍历子文件夹
            print(image_name)
            image = cv2.imread(img_folder + "/" + image_name) #读图片 
            #如果图像上灰度图，请设置第二个参数位0，否则会默认位彩色图
            print(image)
            image = image[:, :, ::-1]
            image = np.vstack((image[:, :, 0], image[:, :, 1], image[:, :, 2]))
            print(image.shape)
            img_list.append(image)
    else:
        print('No data')

    print("img_list",img_list[0])
    #
    for i in range(len(img_list) - 4):
        block = np.vstack(np.array(img_list[i:i + 4]))
        save_name = str(i+1) + '-' + str(i+5)
        np.save("cover/" + save_name, block)
        print(block.shape)

复制