由于报错太长,文章题目写不下了,这里重新说明一下。相信有不少小伙伴在使用 Python
的时候都遇到过以下错误:
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()
那么产生这个错误的原因是什么呢?以及解决方案又是什么呢?
如果我们生成一个如下的一维数组:
[0 1 2 3 4 5 6 7 8 9]
我们想要对其中每一个元素进行判断,凡是小于 5
的元素对应的布尔值为 True
, 大于等于 5
的元素对应的布尔值为 False
。以至于最终可得得到如下结果。
[False False False True True False False False False False]
请看如下代码:
x = np.arange(10)
y = 2 < x < 5
print(y)
"""
result:
Traceback (most recent call last):
File "D:/gitee/ZKWJ_CODE/debug_rect.py", line 52, in <module>
y = 2 < x < 5
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()
"""
可以看到此时的输出结果提示了我们这种错误,这是因为,当我们写成 2 < x < 5
这种形式时,Python
会将 x
中的所有元素看做一个整体,此时的输出结果只会出现一个布尔值。比如:
x = np.arange(10)
# y = (2 < x) * (x < 5) # 这个东西写一个文档
print(x.any())
y = 2 < x.any()
print(y)
y = x.any() < 5
print(y)
y = 2 < x.any() < 5
print(y)
y = 2 < x.all() < 5
print(y)
"""
result:
True
False
True
False
False
"""
可以看到,2 < x.any()
的结果为 False
,这是因为 x.any()
是用来判断数组 x 中是否存在非 0
元素,如果存在,则返回一个布尔值,显然,这里的 x.any()
应该返回 True
的布尔值。又因为在做比较的时候,布尔值 True
会被自动转换为 1.0
。因此,2 < x.any()
的结果为 False
。
同理,x.any() < 5
的结果为 True
,因为 1 < 5
。
后面的同理,这里就不再分析了。关于 .all()
与 .any()
的用法可以参考这篇------np.all() 函数与 np.any() 函数。
对于 x.all()
判断数组 x
中的所有值是否全部不为 0
,显然,这里的 x.all()
值为 False
。因此,2 < x.all() < 5
的值也为 False
。
那么该如何得到我们想要的结果呢?
x = np.arange(10)
y = (2 < x) * (x < 5)
print(y)
"""
result:
[False False False True True False False False False False]
"""
采用上面的写法,我们就可以得到我们想要的结果,由此可得,将数组 x
与单个值进行比较时,Python
会自动对比数组中的每一个值与所给值的大小,而一旦将数组 x
同时与多个值进行比较,Python
就会将数组 x
内的所有值看做一个整体进行比较。
读者可以自行尝试,以下代码也是行不通的。
x = np.arange(10)
y = (2 < x) and (x < 5)
print(y)
y = (2 < x) or (x < 5)
print(y)
如果大家觉得有用,就点个赞让更多的人看到吧~