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

报错信息:

类型错误,只有整型标量数组才能转换成标量索引,但一般问题都不在于你的索引是不是整数。这个报错一般会出现在你想使用一个索引列表去索引另一个列表,即诸如list[index_list]的形式,此时就会出现此报错,因为 index_list 为 List列表类型,不被允许;如果是数组,则不会报错。

解决方法:

将想要索引的列表转换为numpy数组,再对其进行索引
如下:

test_pos_u, test_pos_v = np.array(u)[np.array(eids)[:test_size]], np.array(v)[np.array(eids)[:test_size]] 
train_pos_u, train_pos_v = np.array(u)[np.array(eids)[test_size:]], np.array(v)[np.array(eids)[test_size:]]

你可能感兴趣的:(python,numpy)