python - 如何检查数组是否在 Python 中的数组列表中？

Question

我有一个 3x3 数组列表 l我想检查是否有不同的单个 3x3 数组 a在列表中。我试过这样:

a in l

但由于以下错误而无法执行:

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

我不明白为什么这样的比较会模棱两可(意图很明确)，但无论如何，有没有办法解决我的问题？

Answer 1

Numpy 对你想要做什么感到困惑。您想知道是否a等效于 l 的任何元素或者如果 a是 l 的一个元素.

a  = np.ones((3,3))
b = np.ones((3,3))
l  = [b]


b in l
>>True

a in l
>>
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-17-cf48b78477bf> in <module>()
----> 1 a in l

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

如果您想知道 a在 l是使用 id()功能。

所以

ids = map(id, l)

id(a) in ids
>> False

id(b) in ids
>>True