Python脚本将文件移动到列表目录或测试目录

Question

我是一个能够将我的数据潜入列表目录或测试目录的Python脚本的时刻。我给脚本提供了一个比率，该比例说明了火车/测试之间的比率应该是多少，根据该比率应该随机将文件移动到训练或测试。

ex。如果比例= 0.5，那么我的数据集的一半会在列车中，另一半在测试中。

其他前。如果比率= 0.25，那么75％的数据集将在训练中，其余的则在测试中。

但分工似乎错每次..我想单独的84个文件/显示目录和似乎无法砸金42/42分离 - ..任何暗示我能做什么不同？

下面是代码：

import sys 
import os 
import shutil 
import numpy 
import random 


src = sys.argv[1] 
destination_data = sys.argv[2] 

src_abs = os.path.abspath(src) 
destination_data_abs = os.path.abspath(destination_data) 

src_files = os.listdir(src_abs) 


def copytree(src, dst, symlinks=False, ignore=None, split=0.5): 
    for item in os.listdir(src): 
     s = os.path.join(src, item) 
     d = os.path.join(dst, item) 
     d_test = os.path.join(dst, 'test', item) 
     d_train = os.path.join(dst, 'train', item) 

     print d_test 
     print d_train 
     minmax=0.0, 1.0 
     rand = random.uniform(*minmax) 
     print rand 
     if rand > split: 
      # Inserted into train 
      if os.path.isdir(s): 
       shutil.copytree(s, d_train, symlinks, ignore) 
       print "Copytree used! - TRAIN" 
      else: 
       shutil.copy2(s, d_train) 
       print "Copy 2 used! - TRAIN" 
     else: 
      # Inserted into test 
      if os.path.isdir(s): 
       shutil.copytree(s, d_test, symlinks, ignore) 
       print "Copytree used! - TEST" 
      else: 
       shutil.copy2(s, d_test) 
       print "Copy 2 used! - TEST" 

copytree(src_abs,destination_data_abs,True) 

正在UNIX机器上执行的代码...如果该事项？

Answer 1

你可以取文件列表，shuffle吧，然后分割它就分割比例而言。

import os 
import numpy 

src_files = os.listdir(".") 
n_files = len(src_files) 

split_ratio = 0.5 
split_index = int(n_files * split_ratio) 

numpy.random.shuffle(src_files) 

print src_files[0:split_index] 
print src_files[split_index:] 

Flipping a coin 84 times will result in a "perfect" 42 heads/42 tails with a probability of 0.0868.