如何在列表中查找数组的索引

时间:2014-04-28 20:32:54

标签: python numpy

假设您有一个数组列表。然后你单独指定一个(它也存在于列表中)。你怎么能得到它的索引?

In [185]: y_train.index(Y_test[0])
 ---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-185-c9084f698632> in <module>()
 ----> 1 y_train.index(Y_test[0])
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

在这里,y_train是&#39;大&#39;包含所有内容的列表,Y_test[0]是其元素之一

1 个答案:

答案 0 :(得分:0)

for i in range(0, len(y_train)):
 for j in range(0, len(Y_test)):
     if y_train[i].shape==Y_test[j].shape:
        v= y_train[i]==Y_test[j]
        if v.all():
           print ('i',i)
           print ('j',j)

这找到了我追求的索引。错误来自这样一个事实,即当与多个元素的数组模糊地工作时,不清楚真值应该是什么