python怎么读取文件夹里部分的文件 python中读取文件夹中所有的文件

文章摘要：python怎么读取文件夹里部分的文件 python中读取文件夹中所有的文件

在python中读取文件夹中部分文件的方法 import cv2 import os data_path = […]

在python中读取文件夹中部分文件的方法

import cv2

import os

data_path = '/home/pi/bg/' #需读取文件的路径

img_list = os.listdir(data_path)

save_path = '/home/pi/bg/'

if not os.path.exists(save_path):

os.mkdir(save_path)

for i in range(0, len(img_list)):

print(i)

img_full_path = os.path.join(data_path, img_list[i])

if os.path.exists(img_full_path) and img_full_path.endswith('_label.png'):

label = cv2.imread(img_full_path)

label = label[..., 2]

save_full_path = os.path.join(save_path, img_list[i][:-9] + 'label' + '.png')

cv2.imwrite(save_full_path, label)