pythonnumpy:int数组可以转换为标量索引

请帮我摆脱这个错误，也许，它是重复的，但我无法为我的代码设置它

import pandas as pd
from sklearn.model_selection import KFold

df = pd.read_csv('DATA.txt',delimiter=',')
df.head() 
X= df.COL1,df.COL2
Y=df.COL3
print(X)
print(Y)
cv = KFold(n_splits=2, random_state=10, shuffle=False)
for train_index, test_index in cv.split(X):
    print("Train Index: ", train_index, "\n")
    print("Test Index: ", test_index)
    
X_train, X_test, Y_train, Y_test = X[train_index], X[test_index], Y[train_index], Y[test_index]
print(X_train)
print(Y_train)

错误

---

我认为您可能希望选择的

X

列稍有不同，例如：

X=df[['COL1'，'COL2']]

你能显示所有的打印语句以及已经提供的堆栈跟踪吗。？你认为

X=df.COL1，df.COL2

有什么作用？你能分享

data.txt

@user2357112>中的样本数据吗？我有共享的样本。@JoeFerndz sharednow得到这个：keyrerror:[Int64Index（[0，1，2，3，4，5]，dtype='int64'）]位于[列]“X_-train，X_-test，Y_-train，Y_-test=X[train_-index]，X[test_-index]，Y[train_-index]，Y[test_-index]此代码不工作，而使用

df.iloc[index]

。或者切换到仅使用numpy数组，例如

X=df[['COL1'，COL2']]。值

和

X=df['COL3']。值

。

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-56-8c160cb8bf67> in <module>
----> 1 X_train, X_test, Y_train, Y_test = X[train_index], X[test_index], Y[train_index], Y[test_index]

TypeError: only integer scalar arrays can be converted to a scalar index

9.999999,0.1,1
94.9999,0.1,1
89.9999,0.2,0
99.9999,0.3,0