directory - python : Move same images inside same folder progamtically

Question

Below code successfully creates all folder with same as image names. Now, I want to move images with same name inside same folder. This code creates folders but images were not moved inside folders.

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

Correct one. Just remove .py file

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)