Python|查找最大差异对
有时,我们需要找到产生最大差异的对的具体问题,这可以通过排序并获取列表的第一个和最后一个元素来解决。但是在某些情况下,我们不需要更改列表的顺序并在不使用额外空间的情况下在类似的列表中执行一些操作。让我们讨论可以执行此操作的某些方式。
方法 #1:使用列表理解 + max()
+ combination()
+ lambda
可以使用上述函数的组合来执行此特定任务,其中我们使用列表推导绑定所有功能,使用 max函数来获得最大差异,组合函数在内部找到所有差异,而 lambda函数用于计算差异。
# Python3 code to demonstrate
# maximum difference pair
# using list comprehension + max() + combinations() + lambda
from itertools import combinations
# initializing list
test_list = [3, 4, 1, 7, 9, 1]
# printing original list
print("The original list : " + str(test_list))
# using list comprehension + max() + combinations() + lambda
# maximum difference pair
res = max(combinations(test_list, 2), key = lambda sub: abs(sub[0]-sub[1]))
# print result
print("The maximum difference pair is : " + str(res))
输出 :
The original list : [3, 4, 1, 7, 9, 1]
The maximum difference pair is : (1, 9)
方法 #2:使用列表理解 + nlargest()
+ combination()
+ lambda
该方法不仅可以找到单个最大值,还可以在需要时找到 k 个最大差异对,并使用 nlargest函数而不是 max函数来实现此功能。
# Python3 code to demonstrate
# maximum difference pair
# using list comprehension + nlargest() + combinations() + lambda
from itertools import combinations
from heapq import nlargest
# initializing list
test_list = [3, 4, 1, 7, 9, 8]
# printing original list
print("The original list : " + str(test_list))
# using list comprehension + max() + combinations() + lambda
# maximum difference pair
# computes 2 maximum pair differences
res = nlargest(2, combinations(test_list, 2),
key = lambda sub: abs(sub[0]-sub[1]))
# print result
print("The maximum difference pair is : " + str(res))
输出 :
The original list : [3, 4, 1, 7, 9, 8]
The maximum difference pair is : [(1, 9), (1, 8)]