Skip to main content
 首页 » 编程设计

python之如果条件为真则合并子列表

2025年05月04日12exmyth

例子:

mylist = [['2', '12/12/2016'], ['4', '10/12/2016'], ['5', '08/12/2016'], ['4', '10/12/2016'], ['7', '12/12/2016'], ['10', '12/12/2016'], ['11', '08/12/2016'], ['4', '12/12/2016'], ['10', 'test']] 

子列表中的第一个元素是一个字符串数字
子列表中的第二个元素可以是任何字符串

如果子列表中的第一个元素已经在列表中,我想合并子列表(并删除 double 的字符串)

输出:

newlist = [['2', '12/12/2016'], ['4', '10/12/2016', '12/12/2016'], ['5', '08/12/2016'], ['7', '12/12/2016'], ['10', '12/12/2016', 'test'], ['11', '08/12/2016']] 

如何根据条件获取合并列表?

请您参考如下方法:

如果您希望子列表和子列表中的项目的排序尽可能接近原始顺序,您可以使用 OrderedDict :

from collections import OrderedDict 
 
mylist = [['2', '12/12/2016'], ['4', '10/12/2016'], ['5', '08/12/2016'], 
          ['4', '10/12/2016'], ['7', '12/12/2016'], ['10', '12/12/2016'], 
          ['11', '08/12/2016'], ['4', '12/12/2016'], ['10', 'test']] 
 
res = OrderedDict() 
for s1, s2 in mylist: 
    res.setdefault(s1, OrderedDict())[s2] = True 
 
res = [[k] + list(v) for k, v in res.items()] 
print(res) 

输出:

[['2', '12/12/2016'], ['4', '10/12/2016', '12/12/2016'], ['5', '08/12/2016'],  
 ['7', '12/12/2016'], ['10', '12/12/2016', 'test'], ['11', '08/12/2016']] 

以上构造 OrderedDict,其中键是列表中的第一个元素,值是 OrderedDict 对象。二级字典包含列表中的第二个元素作为键。二级值无关紧要,仅使用 OrderedDict 是因为 Python 标准库没有 OrderedSet。

更新:假设

  1. 子列表有第三个元素,它是两个数字的列表
  2. 结果列表应根据第一个元素的首次出现进行排序
  3. 子列表应仅包含唯一的对,并根据对的第一个数字排序

您可以在第二级使用set 并在构造结果时对对进行排序:

from collections import OrderedDict 
 
mylist = [['1', 'string1', [22,25]], ['4', 'string1', [12,19]],  
          ['4', 'string3', [48,53]], ['8', 'string3', [14,19]], 
          ['4', 'string3', [48,53]]] 
 
res = OrderedDict() 
for s, _, l in mylist: 
    res.setdefault(s, set()).add(tuple(l)) 
 
res = [[k] + sorted(v) for k, v in res.items()] 
print(res) 

输出:

[['1', (22, 25)], ['4', (12, 19), (48, 53)], ['8', (14, 19)]]