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

python

867

2021/3/8 9:08:49

栏目: 编程语言

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

在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)

辰迅云「云服务器」，即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘，价格低至29元/月。点击查看>>

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

最新知识库

相关标签