python:按程序在同一文件夹中移动相同的图像

Question

下面的代码成功创建了所有与图像名称相同的文件夹。现在，我想在同一个文件夹中移动相同名称的图像。此代码创建文件夹，但图像未在文件夹中移动。

import os
import shutil
images = os.listdir(os.getcwd())
print(images)
ids = set(i.split('_')[0] for i in images)  # set removes duplicates
for i in ids:
os.mkdir(i)  # create subdirs
for img in images:
target_dir = img.split('_')[0]
shutil.move(img, target_dir)

Answer 1

更正一个。只需删除.py文件

import os
import shutil
images = os.listdir(os.getcwd())
bad_files = set(["morph.py"])
ids = set(i.split('_')[0] for i in images)  # set removes duplicates
ids.remove("morph.py")
print(ids)
for i in ids:
os.mkdir(i)  # create subdirs
for img in images:
target_dir = img.split('_')[0]
shutil.move(img, target_dir)