python随机划分数据集，留下列表子集以外的数据

在划分数据集时，随机取得训练集，剩下的是测试集。这里有一个麻烦事是训练集是随机的，所以要怎么减呢。我的想法是用集合来减。

import random
lst1 = ['1','2','3','4','5','6'] # 假设这是总的
split_percent = 0.8

train = random.sample(lst1,int(split_percent*len(lst1)))
test = list(set(lst1)-set(train))

print('train:',train)
print('test:',test)

结果：