ValueError:包含多个元素的数组的真值不明确.使用.any()或.all()，后跟TypeError

Question

我想检查一组列表中是否已经存在字典。在if-statement中没有使用any()，我收到了以下错误：

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

但当我使用any()时，会出现以下错误：

TypeError: 'bool' object is not iterable

我不知道哪里出了问题。如有任何帮助或指导，我们将不胜感激。

import numpy as np
compile = []
MY_LIST = [[['Mon'], np.array([4,2,1,3]), ['text'], ['more_text', 0.1]], [['Mon'], np.array([4,2,1,3]), ['text'], ['more_text', 0.1]], [['Tues'], np.array([3, 1, 2, 4]), ['text2'], ['more_text2', 0.2]]]
for group in MY_LIST:
my_dict = {'A': group[0], 'B': group[1], 'C': group[2],
'D': group[3]}
if any(my_dict not in compile):  # Error here
compile.append(my_dict)
print (compile)

预期输出应为：

[{'A': ['Mon'], 'B': array([4, 2, 1, 3]), 'C': ['text'], 'D': ['more_text', 0.1]}, {'A': ['Tues'], 'B': array([3, 1, 2, 4]), 'C': ['text2'], 'D': ['more_text2', 0.2]}]

Answer 1

与大多数类型不同，问题归结为无法单独使用==直接比较两个numpy数组。您需要定义一个自定义函数来比较字典中的两个值，以处理不同的类型：

import numpy as np
def compare_element(elem1, elem2):
if type(elem1) != type(elem2):
return False
if isinstance(elem1, np.ndarray):
return (elem1 == elem2).all()
else:
return elem1 == elem2
result = []
MY_LIST = [
[['Mon'], np.array([4,2,1,3]), ['text'], ['more_text', 0.1]],
[['Mon'], np.array([4,2,1,3]), ['text'], ['more_text', 0.1]],
[['Tues'], np.array([3, 1, 2, 4]), ['text2'], ['more_text2', 0.2]]
]
for group in MY_LIST:
elem_to_append = dict(zip(['A', 'B', 'C', 'D'], group))
should_append = True
for item in result:
if all(compare_element(item[key], elem_to_append[key]) for key in item):
should_append = False
break
if should_append:
result.append(elem_to_append)
print(result)