我有一个 numpy 数组列表(用作堆栈)。现在我想检查一个数组是否已经在列表中。例如,如果它是元组,我会简单地写一些等同于 (1,1) in [(1,1),(2,2)]
的东西。但是,这不适用于 numpy 数组; np.array([1,1]) in [np.array([1,1]), np.array([2,2])]
是一个错误 (ValueError:具有多个元素的数组的真值不明确。使用 a.any() 或 a.all()
)。错误消息在这里没有帮助 AFAIK,因为它指的是直接比较数组。
我很难相信这是不可能的,但我想我缺少了一些东西。
最佳答案
要测试等于a
的数组是否包含在列表my_list
中,请使用
any((a == x).all() for x in my_list)
关于python - Python 列表中的 Numpy.Array?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/5488307/