1

我有比较相邻列表的代码，如果每个列表的第三个元素是 1，则每个列表的第二个元素被添加到它的相邻第二个元素中。然后代码找到这些求和对中的最低值，最后是这个最低的第二个元素一对。但是现在我想忽略每对的第一个元素相同的情况，例如下面的前两个列表。

aList = [[10564, 15, 1], [10564, 13, 1], [10589, 18, 1], [10637, 39, 1], [10662, 38, 1], [10837, 1, 1], [3, 17, 13], [7, 21, 13], [46, 26, 13]]

import sys
from operator import itemgetter
from itertools import tee, izip,ifilter

def pairwise(iterable):
    #"s -> (s0,s1), (s1,s2), (s2, s3), ..."
    a, b = tee(iterable)
    next(b, None)
    return izip(a, b)

def neighbouring_minimum(iterable):
    paired = pairwise(iterable)

    # find minimum pair by second elements summed
    minpair = min(paired, key=lambda pair: pair[0][1] + pair[1][1])  
    return min(minpair, key=itemgetter(1))

filtered = ifilter(lambda x: x[2] == 1, aList)
print neighbouring_minimum(filtered)

我是 lambda 新手，如果我改变

minpair = min(paired, key=lambda pair: pair[0][1] + pair[1][1])

对于下面的代码，这是正确的功能吗，是否有更简洁的方法来编写它？

minpair = min(paired, key=lambda pair: pair[0][1] if (pair[0][0] != pair[1][0]) else pair[0][1] + pair[1][1] )
4

1 回答 1

2

你的逻辑颠倒了；pair[0][1]如果该对的第一个元素不相等，您将返回。反转条件：

minpair = min(paired, key=lambda pair: pair[0][1] if pair[0][0] == pair[1][0] else pair[0][1] + pair[1][1])

但是，请注意，求和对的价值可能比任何单个价值都要大。对于您的示例，您的逻辑将返回15，这绝对低于您为示例输入生成的任何其他邻居总和：

>>> map(lambda pair: pair[0][1] if pair[0][0] == pair[1][0] else pair[0][1] + pair[1][1], pairwise(aList))
[15, 31, 57, 77, 39, 18, 38, 47]

上面的示例向您展示了 lambda 生成的值，并且15将作为最小值获胜。

要忽略第一个元素相同的对，您应该返回float('inf')：

minpair = min(paired, key=lambda pair: pair[0][1] + pair[1][1] if pair[0][0] != pair[1][0] else float('inf'))

float('inf')（正无穷大）总是会高于任何求和的对，所以min()不会考虑那个特定的对。

现在输出是：

>>> map(lambda pair: pair[0][1] + pair[1][1] if pair[0][0] != pair[1][0] else float('inf'), pairwise(aList))
[inf, 31, 57, 77, 39, 18, 38, 47]

18并且将选择总计的对。

于 2013-04-22T09:09:00.273 回答