数据分析

data_preprocessing/combine_sub_class.py

@@ -0,0 +1,25 @@
+import os
+import shutil
+
+
+def combine_dirs(conf):
+    source_root = conf['data']['combine_scr_dir']
+    target_root = conf['data']['combine_dst_dir']
+    for roots, dir_names, files in os.walk(source_root):
+        for dir_name in dir_names:
+            source_dir = os.path.join(roots, dir_name)
+            target_dir = os.path.join(target_root, dir_name.split('_')[0])
+            if not os.path.exists(target_dir):
+                os.mkdir(target_dir)
+            for filename in os.listdir(source_dir):
+                print(filename)
+                source_file = os.sep.join([source_dir, filename])
+                target_file = os.sep.join([target_dir, filename])
+                shutil.copy(source_file, target_file)
+            # print(f"已复制目录 {source_dir} 到 {target_dir}")
+
+
+# if __name__ == '__main__':
+#     source_root = r'scatter_mini'
+#     target_root = r'C:\Users\123\Desktop\scatter-1'
+#     # combine_dirs(conf)