python划分数据集代码

好的，以下是Python划分数据集代码的示例：

import numpy as np

data = np.array([[1, 2], [3, 4], [5, 6], [7, 8], [9, 10]]) # 假设数据集

# 将数据集随机分为训练集和测试集，比例为3:1
np.random.shuffle(data) # 随机打乱数据集
train_data = data[:3] # 取前3条数据作为训练集
test_data = data[3:] # 取剩下2条数据作为测试集

希望以上代码能够帮到你。