我正在寻找一种方法,可以让我检查比较2个列表时缺少的元素。非常类似于该线程,但我想用NumPy Python编写。
import numpy as np
numbers = np.array([1,2,3,4,5,6,7,8,9])
A = np.array([2,5,6,9])
def listComplementElements(list1, list2):
storeResults = []
for i in list1:
for j in list2:
if i != j:
#write to storeResults if 2 numbers are different
storeResults.append(i)
else:
#if numebrs are equal break out of the loop
break
return storeResults
result = listComplementElements(numbers, A)
print(result) #expected result [1,3,4,7,8]
此刻输出看起来像这样:[1、1、1、1、1、3、3、3、3、4、4、4、4、5、6、6、7、7、7、7、7 ,8、8、8、8、9、9、9]
我在做什么错?
答案 0 :(得分:3)
这将为您提供两个数组的互补:
list(set(numbers) - set(A))
[1, 3, 4, 7, 8]
如果您要保留重复的值,则可以使用以下内容:
from collections import Counter
c1 = Counter(numbers)
c2 = Counter(A)
diff = c1 - c2
print(list(diff.elements()))
答案 1 :(得分:1)
这就是你要去的地方
import numpy as np
numbers = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
A = np.array([2, 5, 6, 9])
def listComplementElements(list1, list2):
storeResults = []
for num in list1:
if num not in list2: # this will essentially iterate your list behind the scenes
storeResults.append(num)
return storeResults
result = listComplementElements(numbers, A)
print(result) #expected result [1,3,4,7,8]
打印:
[1、3、4、7、8]
您应该只有一个循环,然后说if not in
才能评估与第二个列表的关系。否则,您将不必要地循环,就像您看到的那样,它将多次输出结果!此外,使用is in
或not in
之类的运算符可以使您的代码比尝试理解许多嵌套循环更易读:)