python报错：only integer scalar arrays can be converted to a scalar index

Posted 2021-07-15 _刘文凯_

如题，在使用python时出现错误：only integer scalar arrays can be converted to a scalar index ; 类似的错误都可以用这个方法解决 交叉验证时错误等

代码

for tr_idx, val_idx in kfold.split(x_train, y_train):
     tr_X_tmp, val_X_tmp, tr_y, val_y = x_train[tr_idx], x_train[val_idx], y_train[tr_idx], y_train[val_idx]
     	...

原因：

数据类型不匹配
x_train， y_train 必须是numpy类型，而不能是list类型

解决：

类型转换

import numpy as np
x_train = np.array(x_train)
y_train = np.array(y_train)
for tr_idx, val_idx in kfold.split(x_train, y_train):
     tr_X_tmp, val_X_tmp, tr_y, val_y = x_train[tr_idx], x_train[val_idx], y_train[tr_idx], y_train[val_idx]
     	...
     	...