将匹配图像移动到单独文件夹的 Python 脚本

Question

我使用 glob (https://docs.python.org/2/library/glob.html) 制作了一个 python 脚本，该脚本将 JPG 图像与其匹配的 XML 注释匹配并将其移动到不同的文件夹（例如，我有 1.jpg、2.jpg、3.jpg、1 .xml, 3.xml) 然后它移动 (1.jpg,1.xml 和 3.jpg,3.xml) 到一个新文件夹。 2.jpg 没有移动，因为没有与此图像匹配的 xml

import os
import glob
import os.path
import shutil

path = os.getcwd()
j=0


os.chdir("path\\to\\folder\\")   
os.mkdir("image_with_xml")     # create a new folder
newpath =  "path\\to\\folder\\"+"image_with_xml" 

while j < len(glob.glob(path+"\\*"))-1:  
    a=glob.glob(path+"\\*")[j]

    b=glob.glob(path+"\\*")[j+1]


    print(a)
    a1 = os.path.splitext(a)[0]
    b1 = os.path.splitext(b)[0]


    if a1==b1:
        j=j+2
        shutil.move(a,newpath)   # move image to new path.
        shutil.move(b,newpath)   # move image to new path.
    else:
        j=j+1

上面的代码适用于移动一些，但不是所有的图像到新文件夹，为了移动剩余的图像，我必须在脚本中创建新文件夹，然后将剩余的图像移动到那里（例如：让我们说我有 100 个 jpg 和 100 个匹配的 XML，然后我第一次运行这个脚本时只有 62 个被移动到新文件夹，第二次我用不同的文件夹名称运行脚本，剩下的 38 个被移动到下一个文件夹）。如何修改脚本，以便将所有匹配 XML 的图像移至一个文件夹？

Answer 1

这个应该能胜任。我创建了两个列表，一个是 xml，一个是 jpg。然后我检查两个列表中是否存在文件名。如果是：移动！

为了便于阅读，我添加了一个新函数来创建列表。

import os
import glob
import shutil

def remove_ext(list_of_pathnames):
    """
    removes the extension from each filename
    """
    return [os.path.splitext(filename)[0] for filename in list_of_pathnames]

path = os.getcwd()
os.chdir("path\\to\\folder\\")   
os.mkdir("image_with_xml")     # create a new folder
newpath = os.path.join("path\\to\\folder\\","image_with_xml") # made it os independent... 

list_of_jpgs = glob.glob(path+"\\*.jpg")
list_of_xmls = glob.glob(path+"\\*.xml")

print(list_of_jpgs, "\n\n", list_of_xmls) #remove

jpgs_without_extension = remove_ext(list_of_jpgs)
xmls_without_extension = remove_ext(list_of_xmls)

print(jpgs_without_extension, "\n\n", xmls_without_extension) #remove

for filename in jpgs_without_extension:
    if filename in xmls_without_extension:
        print("moving", filename) #remove
        shutil.move(filename + '.jpg'), newpath)   # move image to new path.
        shutil.move(filename + '.xml'), newpath)   # move image to new path.