我有问题。 我正在使用train_test_split拆分数据，但无法在该随机数据集中获得测试索引。
这是示例代码：
data = pd.read_csv("data.csv")
X=data.iloc[:,1:-2].values
Y=data.iloc[:,11]
X_train, X_test, Y_train, Y_test = train_test_split(X, Y, test_size = 1/40)
print(Y_test)
这是结果
11244 36
11511 35
18553 17
10901 43
38161 57
35124 39
6154 52
...
13923 30
9523 16
16611 37
26026 137
37327 11
38242 36
13910 34
14343 28
13441 8
26022 30
8981 23
10712 33
33620 5
21631 55
Name: BES_AY, Length: 1000, dtype: int64
我如何才能将结果首先放入数组中？ 像array [0] = 11244一样，array [1] = 11511 ...