划分数据集为训练集和测试集python

可以使用scikit-learn库中的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为随机种子，确保每次划分的结果相同。划分后，X_train和y_train为训练集的特征矩阵和目标变量，X_test和y_test为测试集的特征矩阵和目标变量。