python划分测试集和训练集的代码

以下是一个简单的 Python 代码示例，用于将数据集随机拆分为训练集和测试集：

import random

def split_dataset(dataset, split_ratio):
    train_size = int(len(dataset) * split_ratio)
    train_set = []
    test_set = list(dataset)
    while len(train_set) < train_size:
        index = random.randrange(len(test_set))
        train_set.append(test_set.pop(index))
    return [train_set, test_set]

其中，`dataset` 是一个列表，包含了所有的数据样本，`split_ratio` 表示训练集占整个数据集的比例。函数会将数据集随机拆分为训练集和测试集，并返回一个列表，其中第一个元素是训练集，第二个元素是测试集。可以使用以下代码将数据集拆分为训练集和测试集：

dataset = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
split_ratio = 0.7
train_set, test_set = split_dataset(dataset, split_ratio)
print(train_set)
print(test_set)

输出结果类似于：

[2, 8, 6, 10, 5, 3, 1]
[4, 7, 9]

这表示训练集包含了 70% 的数据样本，测试集包含了 30% 的数据样本。