train_test_split函数中array该如何设置

train_test_split函数中的arrays参数指的是要划分的数据集，可以是一个或多个数组。如果只有一个数组，那么该数组就是样本特征集，并会被划分为训练集和测试集。如果有两个数组，则第一个数组是样本特征集，第二个数组是样本标签集，并且两个数组的行数必须相等。如果有多个数组，则每个数组都必须有相同的行数，且按行同时划分。

例如，如果有一个名为data的DataFrame对象，它包含特征和标签列，可以这样设置：

X = data.drop('label', axis=1)  # 特征列
y = data['label']  # 标签列
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和y同时进行划分，返回训练集和测试集的特征和标签集。