根据python

时间:2018-12-19 08:08:53

标签: python python-2.7

我有2个如下所示的列表

list1=[['a',2,3],['b',4,5],['c',6,7]]
list2=[['a',5],['b',3],['c',8]]

我想根据子列表的条件将此2个列表作为1个列表加入。 如果list1中子列表的第一个元素等于list2中子列表的第一个元素,则list2中子列表的第二个元素应添加到list1

结果应该是

list1=[['a',2,3,5],['b',4,5,3],['c',6,7,8]]

我可以对单个列表项执行以下操作

list1=['a',2,3]
list3=['a',5]

if list1[0]==list3[0]:
   list1.extend(list3[1:])
else:
   list1
print list1

如上所述,在帮助我做子列表时也一样。我无法申请2个循环。

救救我

3 个答案:

答案 0 :(得分:4)

这是您的初始数据:

>>> list1=[['a',2,3],['b',4,5],['c',6,7]]
>>> list2=[['a',5],['b',3],['c',8]]

然后,我们zip您的列表,并使用列表理解来生成新列表:

>>> [ a+b[1:] for a,b in zip( list1, list2 ) if a[0] == b[0] ]

结果:

[['a', 2, 3, 5], ['b', 4, 5, 3], ['c', 6, 7, 8]]

只需一行代码:)

已编辑1

  

如果list2 = [['a',5],['c',8],['b',3]],则无法使用。帮帮我

>>> list1=[['a',2,3],['b',4,5],['c',6,7]]
>>> list2=[['a',5],['c',8],['b',3]]  # <--- other list2

在其他位置使用if

>>> [ a+ ( b[1:] if a[0] == b[0] else [] )  for a,b in zip( list1, list2 ) ]

结果:

[['a', 2, 3, 5], ['b', 4, 5], ['c', 6, 7]]

已编辑2

  

但是list1中的第二个和第三个子列表在list2中具有值,但未按顺序排列为list1。

在zip之前排序。

>>> zipped_sorted_list = zip( sorted(list1), sorted(list2) ) 
>>> [ a+ ( b[1:] if a[0] == b[0] else [] ) for a,b in zipped_sorted_list ]

结果

[['a', 2, 3, 5], ['b', 4, 5, 3], ['c', 6, 7, 8]]

答案 1 :(得分:2)

我建议将主列表转换为字典:

dict1 = { e[0]:e[1:] for e in list1 }
print(dict1)
#=> {'a': [2, 3], 'c': [6, 7], 'b': [4, 5]}

然后遍历list2,将元素添加到字典中:

for e in list2:
  dict1[e[0]].extend(e[1:])
#=> {'a': [2, 3], 'c': [6, 7], 'b': [4, 5]}

如果您需要将其作为列表(待排序):

result = [ [k] + v for k, v in dict1.items() ]
print(result)
#=> [['a', 2, 3, 5], ['c', 6, 7, 8], ['b', 4, 5, 3]]

答案 2 :(得分:0)

list1=[['a',2,3],['b',4,5],['c',6,7]]
list2=[['a',5],['b',3],['c',8]]
#Sort the lists by first element in each sublist
list1.sort(key=lambda x: x[0]) 
list2.sort(key=lambda x: x[0])
#this count is used with the next loop in order to keep track of comparison 
#between list1's sublists with list2's sublists.
count = 0 
for sublist in list1:
    if sublist[0] == list2[count][0]: #comparison logic here
        sublist.extend(list2[count][1:])
    count += 1 #increment the count each loop
print (list1)