根据python中其他列表的值获取最大子列表

Question

希望根据条件获取在指定索引上具有最大值的子列表。

mainlist=[[['RD-2', 'a', 120], ['RD-2', 'b', 125], ['RD-2', 'c', 127]], [['RD-3', 'a', 120], ['RD-3', 'b', 140]], [['RD-5', 'a', 120]],[['RD-7', 'a', 122]]]

和另一个列表

baselist=[['RD-2', 100],['RD-3', 200],['RD-5', 240]]

对于baselist中的每个第一个子元素，我需要来自mainlist的1个完整的子列表，它在第二个索引位置具有最大值。输出应为

flist=[['RD-2', 'c', 127],['RD-3', 'b', 140],['RD-5', 'a', 120]]

帮帮我。

Answer 1

试试这个：

flist = []               
for item in baselist: 
     for i in mainlist:       
         if i[0][0] ==item[0]:               
             flist.append(max(i, key=lambda x:x[2]))

flist将为：

[['RD-2', 'c', 127], ['RD-3', 'b', 140], ['RD-5', 'a', 120]]

您还可以使用itertools.product稍微简化一下for循环：

from itertools import product    

flist = []               
for i, item in product(mainlist,baselist):      
     if i[0][0] ==item[0]:               
          flist.append(max(i, key=lambda x:x[2]))

或者也只有一行：

flist = [max(i, key=lambda x:x[2]) for i, item in product(mainlist,baselist) if i[0][0] == item[0]]

Answer 2

更紧凑，但使用列表理解的结果是相同的：

max_entries = [max(x, key=lambda x:x[2]) for x in mainlist]
flist = [max_entry for max_entry in max_entries for base_entry in baselist if max_entry[0]==base_entry[0]]

Answer 3

下面是一种使用列表理解的方法：

base = list(zip(*baselist))[0]
[max(i, key=lambda x:x[2]) for i in mainlist if i[0][0] in base]
[['RD-2', 'c', 127], ['RD-3', 'b', 140], ['RD-5', 'a', 120]]