Как скопировать определенные файлы c .jpg в другой каталог? - PullRequest
Новая
       9

Как скопировать определенные файлы c .jpg в другой каталог?

2 голосов
/ 10 марта 2020

Я хочу скопировать определенные c jpg файлы в другой каталог, и я не понимаю, почему это не работает? У меня много картинок, и на данный момент я хочу отсортировать по определенным категориям только те, которые имеют начальное имя 15_0_xxx.jpg соответственно 15_1_xxx.jpg 

import cv2

import sys
import os
import shutil 
from os import listdir
from os.path import isfile, join

mypath = "c:/Users/Harum/Desktop/make dir/"
file_names = [ f for f in listdir(mypath) if isfile(join(mypath, f))]

print(str(len(file_names))+ ' images loaded')

cont_M =0
cont_F =0
m_age = "c:/Users/Harum/Desktop/make dir/M_15/"
f_age = "c:/Users/Harum/Desktop/make dir/F_15/"

input_m = []
input_mS =[]
input_fS =[]
input_f = []

def getZeros(number):
    if(number > 10 and number <100):
        return "0"
    if(number < 10):
        return "00"
    else:
        return ""

for i, file in enumerate(file_names):
    if file_names[i][0] == "15_0":
        cont_M +=1
        image = cv2.imread(mypath+file)
        input_m.append(image)
        input_mS.append(0)
        zeros = getZeros(cont_M)
        cv2.imwrite(m_age +"m_age"+str(zeros)+ str(cont_M)+ ".jpg",image)

    if file_names[i][0] == "15_1":
        cont_F +=1
        image = cv2.imread(mypath+file)
        input_f.append(image)
        input_fs.append(1)
        cv2.imwrite(f_age+"F_age"+str(zeros)+ str(cont_M)+ ".jpg",image)

`

1 Ответ

4 голосов
/ 11 марта 2020

Редактировать: забыл копирующую часть. Вы можете использовать shutil для этого

Вы бы лучше с glob и os: 

from shutil import copyfile
import glob
import os

mypath = "c:/Users/Harum/Desktop/make dir/"
destination_path = "c:/Users/Harum/Desktop/copy/"
# using fstrings to add wildcard character to consider all files. You could add a
## file extension after, as in f"{mypath}15_*.jpg"
file_names = glob.glob(f"{mypath}15_*")

# skip the middle to the ifs
# (...)

# removed the enumerate as it doesn't seems like you're using the positional list index

for file in file_names:
    # getting only the filename (with extension)
    file_name = os.path.basename(file)

    # using the str().startswith() to check True or False
    if file_name.startswith("15_0"):
        cont_M +=1
        copyfile(file, f"{destination_path}{file_name}")
        # (...)

    elif file_name.startswith("15_1"):
        cont_F +=1
        copyfile(file, f"{destination_path}{file_name}")
        #(...)
...