划分数据集为训练集和测试集代码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为测试集的特征和标签。