scikit-learnのtrain_test_split関数の引数について

Question

前提・実現したいこと

以下に記すコード中に、train_test_splitの引数にrange(5)というものがありますが、これはどのような操作を指定しているものなのでしょうか。
sklearnの公式サイト等を調べてみましたが、該当の説明を見つけることができませんでした。
初歩的な質問で恐縮ですが、ご教示いただけますと幸いです。

該当のソースコード

from sklearn.model_selection import train_test_split
import string

lettersAtoE = string.ascii_uppercase[:5]
print(lettersAtoE) #ABCDE

train0to4, test0to4, trainAtoE, testAtoE = train_test_split(range(5), lettersAtoE)

print(train0to4)# [0, 1, 2]
print(trainAtoE)# ['A', 'B', 'C']

print(test0to4)# [4, 3]
print(testAtoE)# ['E', 'D']

Answer 1

train_test_splitはarrayを入力すると、分割されたarrayを出力します。

range(5)は[0, 1, 2, 3, 4]というリストに相当するので、これを分割した二つのarrayがそれぞれtrain0to4とtest0to4に代入されます。（例：[0, 1, 2]と[4, 3]）

また、train_test_splitは任意個のarrayを入力にとれるので、例のように2つ入力すると、それぞれを分割したもの（計4個）が返されます。

Comments

[SuzuAya]

>Amakaze様
大変分かりやすいご回答をありがとうございます。おかげさまで理解できました！ありがとうございました。