python划分数据集代码

可以使用sklearn库中的train_test_split函数来划分数据集。该函数可以将数据集随机分为训练集和测试集。

以下是示例代码：

from sklearn.model_selection import train_test_split

# 假设X为特征矩阵，y为目标变量
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

其中，X为特征矩阵，y为目标变量，test_size表示测试集所占的比例，random_state为随机种子，用于保证每次划分的结果都是一样的。上述代码将数据集划分为80%的训练集和20%的测试集。