2

我正在尝试在 2D numpy 数组中搜索特定值，get_above 方法返回字符“初始字符”上方的坐标列表

def get_above(current, wordsearch):
list_of_current_coords = get_coords_current(current, wordsearch)
#print(list_of_current_coords)
length = len(list_of_current_coords)
first_coords = []
second_coords = []
for x in range(length):
    second = list_of_current_coords[x][1]
    new_first = list_of_current_coords[x][0] - 1
    first_coords.append(new_first)
    second_coords.append(second)
combined = [first_coords, second_coords]
above_coords = []
for y in range(length):
    lst2 = [item[y] for item in combined]
    above_coords.append(lst2)   
return above_coords

def search_above(initial_char, target, matrix):
    above_coords = get_above(initial_char, matrix)
    length = len(above_coords)
    for x in range(length):
        if matrix[above_coords[x]] == target:
            print(above_coords[x])
        else:
            print('not found')

调用函数时出现此错误：

ValueError：具有多个元素的数组的真值不明确。使用 a.any() 或 a.all()

任何帮助，将不胜感激！

4

1 回答 1

5

ifValueError 是由语句中的数组比较引起的。

让我们做一个更简单的测试用例：

In [524]: m=np.arange(5)
In [525]: if m==3:print(m)
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-525-de75ce4dd8e2> in <module>()
----> 1 if m==3:print(m)

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()
In [526]: m==3
Out[526]: array([False, False, False,  True, False], dtype=bool)

测试m==3产生一个布尔数组。这不能在if上下文中使用。

any或者all可以将该数组压缩为一个标量布尔值：

In [530]: (m==3).any()
Out[530]: True
In [531]: (m==3).all()
Out[531]: False

所以在

if matrix[above_coords[x]] == target:
        print(above_coords[x])

查看matrix[above_coords[x]] == target，并准确决定如何将其转换为标量 True/False 值。

于 2016-11-26T17:30:20.310 回答