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

最新推荐文章于 2024-03-01 23:13:22 发布

原创最新推荐文章于 2024-03-01 23:13:22 发布 · 302 阅读

CC 4.0 BY-SA版权

本文解决了在使用numpy数组进行索引时遇到的TypeError问题，详细介绍了如何将List转换为numpy数组来避免错误，确保了数据正确划分。

如
for i, (train, test) in enumerate(kf.split(data)):   # 将训练集的?条数据划分为5份
    tra = np.array(data)[train]
    tst = np.array(data)[test]

产生错误：TypeError: only integer scalar arrays can be converted to a scalar index

解决：将List转换为numpy数组

改成：

for i, (train, test) in enumerate(kf.split(data)):   # 将训练集的条数据划分为5份
    tra = np.array(data)[train]
    tst = np.array(data)[test]