简介：如果有一批来的牙科数据有问题，从混在一起的数据中挑出属于这一批的数据
输入：

1. source_occ_dir：需要被挑出的occ点
2. source_4_dir：需要被挑出的点云
3. obj_dir：根据哪个文件夹下的名字去挑数据，比较重要的部分
4. dest_occ_dir：occ点挑到哪里去
5. dest_4_dir：点云挑到哪里去

import os
import shutil

# 定义路径
source_occ_dir = "/data7/haolin/TeethData/RD_3/crown_occ"
source_4_dir = "/data7/haolin/TeethData/RD_3/crown_4_pointcloud"
obj_dir = "/data7/haolin/TeethData/Aidite_Crown_Dataset_forth_scale/tooth_crown_watertight"
dest_occ_dir = "/data7/haolin/TeethData/Aidite_Crown_Dataset_forth_scale/crown_occ"
dest_4_dir = "/data7/haolin/TeethData/Aidite_Crown_Dataset_forth_scale/crown_4_pointcloud"

def move_npz_files():


    # 遍历obj目录中的所有文件
    for obj_file in os.listdir(obj_dir):
        if obj_file.endswith(".obj"):
            base_name = os.path.splitext(obj_file)[0]
            
            # 源文件路径
            npz_occ_source = os.path.join(source_occ_dir, f"{base_name}.npz")
            npz_4_source = os.path.join(source_4_dir, f"{base_name}.npz")
            
            # 目标文件路径
            npz_occ_dest = os.path.join(dest_occ_dir, f"{base_name}.npz")
            npz_4_dest = os.path.join(dest_4_dir, f"{base_name}.npz")

            # 移动crown_occ的npz文件
            if os.path.exists(npz_occ_source):
                shutil.move(npz_occ_source, npz_occ_dest)
                print(f"Moved {npz_occ_source} to {npz_occ_dest}")
            
            # 移动crown_4_pointcloud的npz文件
            if os.path.exists(npz_4_source):
                shutil.move(npz_4_source, npz_4_dest)
                print(f"Moved {npz_4_source} to {npz_4_dest}")

if __name__ == "__main__":
    move_npz_files()