找到第一个元素是不大于给定数字的最大值的子列表

Question

我有一个子列表，每个子列表的第一个元素是一个数字。我想找到第一个元素是不大于给定数字的最大值的子列表。我想知道如何实现？

例如，我想在列表a 中查找子列表，使其第一个元素是不大于3 的最大元素。子列表是[2,'b']。

>>> a=[[5,'d'] ,[1,'a'],[4,'c'],[2,'b'] ]
>>> a = sorted(a)
>>> a
[[1, 'a'], [2, 'b'], [4, 'c'], [5, 'd']]
>>> [3>=x for [x,_] in a]
[True, True, False, False]
>>> a[1]
[2, 'b']

感谢和问候！

Answer 1

>>> a=[[5,'d'] ,[1,'a'],[4,'c'],[2,'b'] ]
>>> max(filter(lambda sl: sl[0]<3, a), key=lambda sl: sl[0])
[2, 'b']

分解：

1) 使用filter产生符合sl[0]<3条件的列表列表的子列表：

>>> filter(lambda sl: sl[0]<3, a)
[[1, 'a'], [2, 'b']]

1.a) 你也可以使用列表推导：

>>> [sl for sl in a if sl[0]<3]
[[1, 'a'], [2, 'b']]

2) 然后使用键函数找到该子集列表的max：

>>> max([[1, 'a'], [2, 'b']], key=lambda sl: sl[0])
[2, 'b']

3) 合并 -- 一行 -- 不排序 -- 快乐...

Answer 2

def grab_max_pair(lst_of_pairs, num):
    result = None
    for pair in lst_of_pairs:
        if result and pair[0] <= num:
            if pair[0] > result[0]:
                result = pair
        elif pair[0] <= 3:
            result = pair
    return result

a=[[5,'d'] ,[1,'a'],[4,'c'],[2,'b'] ]    
print grab_max_pair(a, 3)  # prints [2,b]

Answer 3

您可以使用类似以下的列表推导：

a = # define your list here
new_list = [list for list in a if list[0] < 4]  # only grab sub-lists that meet your criterion
new_list = sorted(new_list)  # sort them now (shorter list)
result = new_list[0]  # grab the first result

如果这是你经常做的事情，你可以把它全部扔到一个函数中：

def get_first(my_list, criterion=4):
    new_list = [list for list in my_list if list[0] < criterion]
    new_list = sorted(new_list)
    return new_list[0] if new_list is not None else None  # avoid a crash if new_list[0] does not have meaning

然后，您可以在导入您放置在其中的任何模块之后，或在您的环境中定义它之后，从 Python 中调用它（有或没有标准值，默认值为 4）：

>> my_list = # define your list here
>> smallest_match = get_first(my_list)