python随机划分训练集——python取list中不包含的元素_python list 不包含

仅作为记录，大佬请跳过。

python随机划分训练集

用test_set = random.sample(source_lst, int(0.2*len(source_lst)))

参考

传送门

python取list中不包含的元素

用train_set = [x for x in source_lst if x not in test_set]

全部程序

'''source中所有图块'''
source_dir = '/data_sda/dataset/source'
source_lst = glob.glob(os.path.join(source_dir, '*.tif'))

'''随机取训练集80%，测试集20%'''
import random
random.seed(0)
test_set = random.sample(source_lst, int(0.2*len(source_lst)))

'''取list中不包含的元素'''
train_set = [x for x in source_lst if x not in test_set]
1
2
3
4
5
6
7
8
9
10
11