gpt4 book ai didi

python - 比较 Numpy 数组会引发 a.any() 或 a.all() 错误

转载 作者:行者123 更新时间:2023-11-28 22:20:40 32 4
gpt4 key购买 nike

我正在尝试使用以下代码来检查是否在另一个“b”列表中找到了 numpy 数组的元素,但我收到以下错误

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

我试着查找它,但无法让它工作

如何在本例中使用 a.all() 或其他方法,如 numpy.logical_and

import numpy as np
a=np.array([[0,0,0,0,1,0,1,1,0],[0,0,0,0,0,1,1,1,0],[0,0,0,0,0,0,1,1,1]])
b=[]

for item in a :
if item not in b:
b.append(item)`

最佳答案

不要循环:

b = np.unique(a, axis=0)

这速度快了几个数量级,也更清晰。

关于python - 比较 Numpy 数组会引发 a.any() 或 a.all() 错误,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/48768027/

32 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com